59.螺旋矩阵II

59. 螺旋矩阵 II
题意
给你一个正整数 n ,生成一个包含 1 到 n 的 2 次方 的所有元素,且元素按顺时针顺序螺旋
排列的 n x n 正方形矩阵 matrix 。
难度
中等
示例
示例 1:
1668213829400-c5c1a443-27bb-4b25-a641-592356f021d8.jpeg
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
示例 2:
输入:n = 1
输出:[[1]]
分析
这题和 054.螺旋矩阵 相差无几,都是需要 螺旋 遍历整个二维数组,只不过上一次是依次
取数字,而这次,我们需要填充数字而已。
还是一样的按照顺时针遍历外层,然后逐步缩小外层,这题便能够解决了。
class Solution{
public int[][] generateMatrix(int n) {
int[][] resMatrix = new int[n][n];
if(n == 0)
return resMatrix;
int up = 0,down = n - 1;
int lef = 0,rig = n - 1;
int cnt = 0;
for(;;){
for(int i = lef;i <= rig;i++)
resMatrix[up][i] = ++cnt;
if(++up > down) break;
for(int i = up;i <= down;i++)
resMatrix[i][rig] = ++cnt;
if(--rig < lef) break;
for(int i = rig;i >= lef;i--)
resMatrix[down][i] = ++cnt;
if(--down < up) break;
for(int i = down;i >= up;i--)
resMatrix[i][lef] = ++cnt;
if(++lef > rig) break;
}
return resMatrix;
}
}
1668213839917-6b24e13f-e145-4c32-ab4e-52d542ac6b1a.png
总结
本题与 054.螺旋矩阵 的思路以及代码大体上没有差别,都是顺时针遍历然后缩小边界,
再填充/获取相应位置的数据即可。
力扣链接:https://leetcode.cn/problems/spiral-matrix-ii/