给定一个正整数 n,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。

示例:

输入: 3
输出:
[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]

 

思路:

和上一篇博客一个思路,也是转圈,不过打印成了赋值,知道n也就知道矩阵的行和列了。

int index = 1;
	public int[][] generateMatrix(int n) {

		int[][] arr = new int[n][n];
		int a = 0, b = 0;
		int c = arr.length - 1, d = arr[0].length - 1;
		while (a <= c && b <= d && index <= n) { // 这个循环用来控制边框
			helper(arr, a++, b++, c--, d--);
		}
		return arr;
	}

	private void helper(int[][] arr, int a, int b, int c, int d) {
		// 判断同行的情况
		if (a == c) {
			for (int i = b; i <= d; i++) {
				arr[a][i] = index++;
			}
		} else if (b == d) { // 判断同列的情况
			for (int i = a; i <= c; i++) {
				arr[i][b] = index++;
			}
		} else {
			int curRow = a;
			int curCol = b;
			// 水平向右方向
			while (curCol != d) {
				arr[curRow][curCol++] = index++;
			}
			// 竖直向下方向
			while (curRow != c) {
				arr[curRow++][curCol] = index++;
			}
			// 水平向左方向
			while (curCol != b) {
				arr[curRow][curCol--] = index++;
			}
			// 竖直向上方向
			while (curRow != a) {
				arr[curRow--][curCol] = index++;
			}
		}
	}