螺旋矩阵II

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

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

方法:模拟

算法思路:

  1. 定义当前上下左右边界t,b,l,r,初始值num = 1,迭代终止值tar = n * n
  2. 按顺序填入空矩阵,填完一条边时,边界要收缩。
  3. num <= tar作为循环条件,而不是l < r || t < b,以防n为奇数时,矩阵中心数字无法填充。

代码实现:

class Solution {
public int[][] generateMatrix(int n) {
int[][] ans = new int[n][n];
int l = 0, t = 0, r = n - 1, b = n - 1;
int tar = n * n;
int num = 1;
while(num <= tar){
for(int i = l; i <= r; i++){
ans[t][i] = num++;
}
t++;
for(int i = t; i <= b; i++){
ans[i][r] = num++;
}
r--;
for(int i = r; i >= l; i--){
ans[b][i] = num++;
}
b--;
for(int i = b; i >= t; i--){
ans[i][l] = num++;
}
l++;
}
return ans;
}
}

参考

Krahets题解