题目描述
给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
示例 1:
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
示例 2:
输入:n = 1
输出:[[1]]
解题思路
此题相对上一题leetcode54更简单,因为边界更可控,是正方形,可以从外圈向中央逼近,重点是边界的处理,避免重复
代码实现
java
class Solution {
public int[][] generateMatrix(int n) {
int[][] r = new int[n][n];
int lr = 0, rr = n -1, sum = n* n;
fill(r, lr, rr, 1, sum);
return r;
}
private void fill(int[][] r, int lr, int rr, int n, int sum) {
if(n > sum) return;
if(lr > rr) return;
for (int i = lr; i <= rr; i++) {
r[lr][i] = n;
n++;
}
for (int i = lr+1; i <= rr; i++) {
r[i][rr] = n;
n++;
}
for (int i = rr-1; i >= lr; i--) {
r[rr][i] = n;
n++;
}
for (int i = rr-1; i > lr; i--) {
r[i][lr] = n;
n++;
}
fill(r, lr+1, rr-1, n, sum);
}
}