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; } }