LeetCode刷题笔记 54. 螺旋矩阵 59. 螺旋矩阵 II

题目要求

给定一个包含 m x n 个元素的矩阵（m 行, n 列），请按照顺时针螺旋顺序，返回矩阵中的所有元素。

示例 1:

输入:
[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]
输出: [1,2,3,6,9,8,7,4,5]

示例 2:

输入:
[
  [1, 2, 3, 4],
  [5, 6, 7, 8],
  [9,10,11,12]
]
输出: [1,2,3,4,8,12,11,10,9,5,6,7]

54. 螺旋矩阵

YouLookDeliciousC

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        vector <int> ans;
        if(matrix.empty()) return ans; //若数组为空，直接返回答案
        int u = 0; //赋值上下左右边界
        int d = matrix.size() - 1;
        int l = 0;
        int r = matrix[0].size() - 1;
        while(true)
        {
            for(int i = l; i <= r; ++i) ans.push_back(matrix[u][i]); //向右移动直到最右
            if(++ u > d) break; //重新设定上边界，若上边界大于下边界，则遍历遍历完成，下同
            for(int i = u; i <= d; ++i) ans.push_back(matrix[i][r]); //向下
            if(-- r < l) break; //重新设定有边界
            for(int i = r; i >= l; --i) ans.push_back(matrix[d][i]); //向左
            if(-- d < u) break; //重新设定下边界
            for(int i = d; i >= u; --i) ans.push_back(matrix[i][l]); //向上
            if(++ l > r) break; //重新设定左边界
        }
        return ans;
    }
};

59. 螺旋矩阵 II

仿I

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n,vector<int>(n));
        int l=0,u=0,r=n-1,d=n-1,num=1;
        while(true){
            for(int i=l;i<=r;i++) res[u][i]=num++;
            if(++u>d) break;
            for(int i=u;i<=d;i++) res[i][r]=num++;
            if(--r<l) break;
            for(int i=r;i>=l;i--) res[d][i]=num++;
            if(--d<u) break;
            for(int i=d;i>=u;i--) res[i][l]=num++;
            if(++l>r) break;
        }
        return res;
    }
};

微调

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n,vector<int>(n));
        int n2=pow(n,2);
        int l=0,u=0,r=n-1,d=n-1,num=1;
        while(num<=n2){
            for(int i=l;i<=r;i++) res[u][i]=num++;
            ++u;
            for(int i=u;i<=d;i++) res[i][r]=num++;
            --r;
            for(int i=r;i>=l;i--) res[d][i]=num++;
            --d;
            for(int i=d;i>=u;i--) res[i][l]=num++;
            ++l;
        }
        return res;
    }
};