leetcode 54 59 螺旋矩阵I和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]

主要思路，通过设置上下左右四个标志位，对矩阵进行顺时针的读取。

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        vector<int> res;
        if(matrix.empty() || matrix[0].empty()) 
            return res;
        int m=matrix.size(),n=matrix[0].size();
        int down=0,up=m-1,left=0,right=n-1;
        while(down<=up && left<=right){
            for(int i=left;i<=right;i++) 
                res.push_back(matrix[down][i]);
            for(int i=down+1;i<=up;i++) 
                res.push_back(matrix[i][right]);
            if(down<up){//单行的情况
                for(int i=right-1;i>=left;i--) 
                    res.push_back(matrix[up][i]);
            }
            if(left<right){//单列的情况
                for(int i=up-1;i>=down+1;i--) 
                    res.push_back(matrix[i][left]);
            }
            down++;
            up--;
            left++;
            right--;
        }
        return res;
    }
};

59题是54题的反转，给定数据范围，构建一个顺时针的矩阵。

给定一个正整数 n，生成一个包含 1 到 n2 所有元素，且元素按顺时针顺序螺旋排列的正方形矩阵。

示例:

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

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        if(n==0) 
            return {{}};
        vector<vector<int>> res(n,vector<int>(n,0));
        int down=0,up=n-1,left=0,right=n-1;
        int itm=1;
        while(left<=right && down<=up){
            for(int i=left;i<=right;i++)//行不变列变，第一行
                res[down][i]=itm++;
            for(int i=down+1;i<=up;i++)//行变列不变，最后一列
                res[i][right]=itm++;
            for(int i=right-1;i>=left;i--)//行不变列变，最后一行
                res[up][i]=itm++;
            for(int i=up-1;i>=down+1;i--)//行变列不变，第一列
                res[i][left]=itm++;
            down++;
            up--;
            left++;
            right--;
        }
        return res; 
    }
};