Little useless-useful R functions – Return matrix elements in spiral order
[This article was first published on R – TomazTsql, and kindly contributed to R-bloggers]. (You can report issue about the content on this page here)
Want to share your content on R-bloggers? click here if you have a blog, or here if you don't.
Want to share your content on R-bloggers? click here if you have a blog, or here if you don't.
Another one from the Leetcode challenge. This time, get the elements (single values) from the matrix in a spiral order with a starting position of [1,1].
So, the basic idea is to retrieve a vector of elements from a matrix in the following order:
In this order, we get the vector with the following elements:
So, we need to create the boundaries, count the rows | columns and return a vector of elements. Here is the useless function:
## return elements from matrix in spiral order matrix_spiral <- function(mat) { mr <- dim(mat)[1] nc <- dim(mat)[2] total_len <- mr*nc #path #TRUE -> visited; FALSE -> unvisited visit <- matrix(FALSE, nrow=mr, ncol=nc) #helper variables gor <- 1 dol <- mr levo <- 1 desno <- nc res <- vector() while (length(res) < total_len) { for (i in levo:nc) { if (!visit[gor, i]) { res <- c(res, mat[gor,i]) visit[gor, i] <- TRUE } } gor <- gor + 1 for (i in gor:mr) { if (!visit[i, desno]) { res <- c(res, mat[i,desno]) visit[i, desno] <- TRUE } } desno <- desno - 1 if (gor <= dol) { for (i in desno:levo) { if (!visit[dol, i]) { res <- c(res, mat[dol,i]) visit[dol, i] <- TRUE } } dol <- dol - 1 } if (levo <= desno) { for (i in dol:gor) { if (!visit[i, levo]) { res <- c(res, mat[i,levo]) visit[i, levo] <- TRUE } } levo <- levo + 1 } } return(res) }
As always, the complete code is available on Github in Useless_R_function repository. The sample file in this repository is here (filename: Spiral_matrix.R). Check the repository for future updates.
Happy R-coding and stay healthy!
To leave a comment for the author, please follow the link and comment on their blog: R – TomazTsql.
R-bloggers.com offers daily e-mail updates about R news and tutorials about learning R and many other topics. Click here if you're looking to post or find an R/data-science job.
Want to share your content on R-bloggers? click here if you have a blog, or here if you don't.