题目:
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了。
AC代码:
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;
}
}