LeetCode 59 螺旋矩阵II

题目大意

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

示例:

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

解题思路

本题因为生成矩阵是正方形，所以边界相对好判断，设置四个边界不断进行缩小即可。还有就是初始通过resize对数组控制大小

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