题目:
给定一个正整数 n,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3
输出:
[
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ]
]
解题思路:
设置这样四个变量:
a:左上角,往数组右边走(行),
b:左上角,往数组下边走(列),
c:右下角,往数组上边走(行),
d:右下角,往数组左边走(列),
这样就形成了一个矩形先遍历a行b到d-1列,然后遍历d列a到c-1行,然后是c行d到b-1列,最后是b列c到a-1行,,,这样类推,遍历完外层,再就将四脚的坐标向内的对角线移动,再按照上面的规则继续遍历,在遍历的过程中同时给数组赋值即可。
代码实现:
class Solution {
public int[][] generateMatrix(int n) {
int[][] ans = new int[n][n];
int a = 0;//行
int b = 0;//列
int c = n-1;//行
int d = n-1;//列
int k = 1;
while(a <= c && b <= d){
if(a == c){//只有一行
for(int i=b;i<=d;i++){
ans[a][i] = k++;
}
break;
}
if(b == d){//只有一列
for(int i=a;i<=c;i++){
ans[i][d] = k++;
}
break;
}
for(int i=b;i<d;i++){//首行
ans[a][i] = k++;
}
for(int i=a;i<c;i++){//尾列
ans[i][d] = k++;
}
for(int i=d;i>b;i--){//尾行
ans[c][i] = k++;
}
for(int i=c;i>a;i--){//首列
ans[i][b] = k++;
}
++a;
++b;
--c;
--d;
}
return ans;
}
}
复杂度分析:
- 时间复杂度: O ( n 2 ) O(n^2) O(n2)
- 空间复杂度: O ( n 2 ) O(n^2) O(n2)