Given an integer n, generate a square matrix filled with elements from 1 to n2 in spiral order.
For example,
Given n = 3,
[ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ] ]
給定一個整數 n,生成一個正方形矩陣。矩陣中包含著從1到n2 這些元素,並且以螺旋順序排列。
比如,給定n = 3,
你應該返回如下的矩陣:
[ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ] ]
和題目《Spiral Matrix》類似,螺旋形的放置,去轉就好。
這裡需要說明的是最後的情況,《Spiral Matrix》中的矩陣形狀不一定是正方形,因此轉到最後剩余的元素可能是一個,也可能是一列或者一行。
本題中已經明確說明了生成的方形的矩陣,因此最終的剩余元素只可能為一個,代碼中考慮這種情況就OK了。
public class Solution
{
public int[][] generateMatrix(int n)
{
int startX = 0;
int startY = 0;
int endX = n - 1;
int endY = n - 1;
int[][] matrix = new int[n][n];
int startValue = 1;
while (startX <= endX)
{
startValue = fillNumber(startX, startY, endX, endY, matrix, startValue);
startX++;//外圈遍歷完,遍歷內圈
startY++;
endX--;
endY--;
}
return matrix;
}
public int fillNumber(int startX, int startY, int endX, int endY, int[][] matrix, int startV)
{
if (startX == endX)
{
matrix[startX][startY] = startV;
return -1;
}
for (int i = startY; i <= endY; i++) //自左向右
{
matrix[startX][i] = startV;
startV++;
}
for (int i = startX + 1; i <= endX; i++) //自上而下
{
matrix[i][endY] = startV;
startV++;
}
for (int i = endY - 1; i >= startY; i--) //自右而左
{
matrix[endX][i] = startV;
startV++;
}
for (int i = endX - 1; i >= startX + 1; i--)//自下而上
{
matrix[i][startY] = startV;
startV++;
}
return startV;
}
}