题目:
给定一个正整数 n,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3 输出: [ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ] ]
思路:直接看了网上大神的代码和leetcode上最快的代码。先把这两个程序看懂。
代码:
public class Solution {
public int[][] generateMatrix(int n) {
int[][] result = new int[n][n];
int layer;
int k;
for (int i = 0; i < n; i++)
{
for (int j = 0; j < n; j++)
{
layer = layer(i, j, n); // 当前坐标外有几层
// n * n - layer * layer外围层使用的最后一个数字(也是最大的)
// 坐标所在的当前层使用的第一个数字
k = n * n - (n - 2 * layer) * (n - 2 * layer) + 1;
if (i == layer) { // 情况一、坐标离上边界最近
result[i][j] = k + j - layer;
} else if (j == n - layer - 1) { // 情况二、坐标离右边界最近
result[i][j] = k + (n - 2 * layer - 1) + i - layer;
} else if (i == n - layer - 1) { // 情况三、坐标离下边界最近
result[i][j] = k + 3 * (n - 2 * layer - 1) - (j - layer); }
else { // 情况三、坐标离左边界最近
result[i][j] = k + 4 * (n - 2 * layer - 1) - (i - layer);
}
}
}
return result; }
/**
* 在一个n*n的矩阵中,计算(x,y)坐标外有多少层,坐标从0开始计算
*
* @param x 横坐标
* @param y 纵坐标
* @param n 矩阵大小
* @return 坐标外的层数
*/
public int layer(int x, int y, int n)
{ x = x < n - 1 - x ? x : n - 1 - x; // 计算横坐标离上下边界的最近距离
y = y < n - 1 - y ? y : n - 1 - y; // 计算纵坐标离左右边界的最近距离
return x < y ? x : y; // 较小的值为坐标的外围层数
}
}
执行最快的代码:
按照上,右,下,左的顺序形成矩阵
class Solution {
public int[][] generateMatrix(int n) {
int[][] matrix = new int[n][n];
int len = n;
int row = 0, col = 0;
int temp = 1;
int start = 0 ;
while (len >= start*2) {
for (int i = start; i < len - 1 - start ; i++) {
matrix[row][col++] = temp++;
}
for (int i = start; i < len - 1 - start ; i++) {
matrix[row++][col] = temp++;
}
for (int i = start; i < len - 1- start; i++) {
matrix[row][col--] = temp++;
}
for (int i = start; i < len - 1- start; i++) {
matrix[row--][col] = temp++;
}
row++;
col++;
start++;
}
if (n % 2 != 0){
matrix[n/2][n/2] = temp;
}
return matrix;
}
}