题目 59. 螺旋矩阵 II
给定一个正整数 n,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3
输出:
[
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ]
]
1public class GenerateMatrix {
2
3 public static void main(String[] args) {
4 int[][] res = generateMatrix(4);
5 for (int i = 0; i < res.length; i++) {
6 for (int j = 0; j < res[i].length; j++) {
7 System.out.print(res[i][j] + " ");
8 }
9 System.out.println();
10 }
11 }
12
13 public static int[][] generateMatrix(int n) {
14 int[][] res = new int[n][n];
15 //开始位置
16 int startX = 0, startY = 0;
17 //循环次数
18 int loop = n / 2;
19 //如果循环次数是奇数,那么会存在中心点,中心点是最后赋值,中心点位置
20 int mid = n / 2;
21 //每条边边界,每循环一圈就加2,因为进内圈需要加1,同时边界向内收缩1
22 int offset = 1;
23 //初始赋值
24 int count = 1;
25 //原则 左闭又开
26 // 上面 从左右导游
27 // 右边 从上到下
28 // 下面 从右到在
29 // 左边 从下到上
30 while ((loop--) > 0) {
31 int i = startX;
32 int j = startY;
33
34 // 上面 从左右导游
35 for (; j < startY + n - offset; j++) {
36 res[startX][j] = count++;
37 }
38
39 // 右边 从上到下
40 for (; i < startX + n - offset; i++) {
41 res[i][j] = count++;
42 }
43
44 // 下面 从右到在
45 for (; j > startY; j--) {
46 res[i][j] = count++;
47 }
48
49 // 左边 从下到上
50 for (; i > startX; i--) {
51 res[i][j] = count++;
52 }
53
54 startX++;
55 startY++;
56 offset += 2;
57
58 }
59
60 if (n % 2 == 1) {
61 res[mid][mid] = count;
62 }
63
64 return res;
65 }
66}