leetcode 54. 螺旋矩阵 及 59. 螺旋矩阵 II

54. 螺旋矩阵

问题描述

给定一个包含 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]

问题分析

这道题也是从最外层往最内层循环，定义四个边界up,down,left,right记录目前层数的位置即可。

代码

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        int m = matrix.size();
         vector<int> ans;
        if(m==0)return ans;
        int n = matrix[0].size();
        ans.resize(m*n);
        int index = 0,up = 0,down = m - 1,left = 0,right = n - 1,i;
        while(1)
        {
            for(i = left; i <= right; ++i)ans[index++] = matrix[up][i];//从左往右遍历
            if(++up > down)break;
            for(i = up; i <= down; ++i)ans[index++] = matrix[i][right];//从上往下遍历
            if(--right < left)break;
            for(i = right; i >= left; --i)ans[index++] = matrix[down][i];//从右往左遍历
            if(--down < up)break;
            for(i = down; i >= up; --i)ans[index++] = matrix[i][left];//从下往上遍历
            if(++left > right)break;
        }
        return ans;
    }
};

59. 螺旋矩阵 II

问题描述

给定一个正整数 n，生成一个包含 1 到 n2 所有元素，且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3
输出:
[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]

代码

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> ans;
        if(n <= 0) return ans;
        ans.resize(n);
        int i,j,up = 0,down = n-1,left = 0,right = n-1,var = 0;
        for(i = 0;i < n; i++)ans[i].resize(n);
       while(true)
       {
           for(i = left;i <= right; ++i)ans[up][i] = ++var;
           if(++up > down)break;
           for(i = up; i <= down; ++i)ans[i][right] = ++var;
           if(--right < left)break;
           for(i = right; i >= left; --i)ans[down][i]=++var;
           if(--down < up)break;
           for(i = down; i >= up; --i)ans[i][left] = ++var;
           if(++left > right)break;
       }
       return ans;
    }
};