54. Spiral Matrix 题解

54. Spiral Matrix  题解

题目描述：

Given a matrix of m x n elements (m rows, n columns), return all elements of the matrix in spiral order.

For example,
Given the following matrix:

[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]

You should return [1,2,3,6,9,8,7,4,5].

题目链接：54. Spiral Matrix

算法描述：

             由题意知，给定一个矩阵，我们将采用 “旋转” 的遍历方式对这个矩阵中的元素进行遍历，将遍历到的矩阵元素存入结果数组并返回。

         “旋转” 遍历即从矩阵第一个元素开始，围绕着矩阵的外围，从左上到右上，从右上到右下，从右下到左下，再从左下到左上，依次进行。

          取到原矩阵的行数 rows 和列数 cols ，仔细观察其遍历的坐标，可以发现每一次 “旋转” 起始坐标为（i，i），即对角线元素，按照上面说明的遍历顺序，向右遍历直至（i，cols-1-i），再向下遍历至 （rows-1-i，cols-1-i），再向左遍历至（rows-1-i，i），再向上遍历至（i，i+1），这样一次遍历结束。

          遍历的轮数为 ( min(rows, cols) + 1) / 2。需要注意，当从右下到左下遍历时如果 rows-1-i == i 并且当从左下到左上时如果 cols -1-i == i ，则不需要再重复遍历。

代码：

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int> > &matrix) {
        vector<int> ans;
        if(matrix.empty() || matrix[0].empty()){
            return ans;
        }
        
        int rows = matrix.size();
        int cols = matrix[0].size();
        int nums = (min(rows,cols) + 1) / 2;
        
        for(int i=0; i<nums; i++){
            
            for(int j=i; j<cols-i; j++){
                ans.push_back(matrix[i][j]);
            }
            
            for(int j=i+1; j<rows-i; j++ ){
                ans.push_back(matrix[j][cols-i-1]);
            }
            
            if(rows-1-i > i){
                for(int j=cols-1-i-1; j>=i; j--){
                    ans.push_back(matrix[rows-1-i][j]);
                }
            }
            
            if(cols-1-i > i){
                for(int j=rows-1-i-1; j>i; j--){
                    ans.push_back(matrix[j][i]);
                }
            }
        }
        
        return ans;

    }
};