携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第1天,点击查看活动详情
一、题目描述:
给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
实例1:
输入: n = 3
输出: [[1,2,3],[8,9,4],[7,6,5]]
解释: n = 3 代表是由数字1到9组成了一个 int[3][3] 的数组
实例2:
输入: n = 1
输出: [[1]]
提示: 1 <= n <= 20
二、思路分析
模拟填入矩阵的过程:
- 从上行左到右
- 从右侧上到下
- 从下行右到左
- 从左侧下到上
使用二分法判断填入矩阵的循环次数, 三次有两圈, 六次有三圈, 九次有五圈. 这里要注意的是最后要加个判断, 如果是单数最中间的数字要加进去
采用左闭右开顺序添加数字, 避免后期混乱容易出错, 这里借用卡尔的一张图
三、java代码
public int[][] generateMatrix(int n) {
int loopNum = 0;
int[][] res = new int[n][n];
int start = 0;
int count = 1;
int i,j;
while(loopNum++ < n /2){
for (i = start; i< n - loopNum;i++){
res[start][i] = count;
count++;
}
for (j = start ; j < n-loopNum; j++){
res[j][i] = count;
count++;
}
for (; i >= loopNum;i--){
res[j][i] = count;
count++;
}
for (; j >= loopNum; j--){
res[j][start] = count;
count++;
}
start ++;
}
if (n % 2 == 1){
res[start][start] = count;
}
return res;
}
四、总结
最开始看着还是比较难的, 但是真正的实践起来之后就也还好, 中间碰到一点小问题都是边界没有找好.