54
這一題主要在考對array index的敏感度,先看一下範例,他提供的matrix是
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
應該要輸出的結果是[1,2,3,6,9,8,7,4,5]。所以依序是輸出:
matrix[0][0]
matrix[0][1]
matrix[0][2]
matrix[1][2]
matrix[2][2]
matrix[2][1]
matrix[2][0]
matrix[0][1]
matrix[1][1]
有了這一個概念之後,大概知道是有個index輸出的規則,不過距離寫成程式碼還有一段差距。這個差距主要是不知道詳細要怎麼跑index。直接去感受一下這一題的本質,會有一種不斷向內縮的感覺,不斷向內縮就會對應到兩種作法,一種是每輸出一圈就把剩下的matrix丟進一個recursive function再處理下一個內圈,另一種做法就是設立row跟col的輸出起點跟終點,然後每輸出完一個row或一個col就把這個起終點向內縮。
兩個做法在很高的抽象層次想起來好像都可以實現,可是如果往下跳到程式碼的抽象層,會發現要把matrix的外圈去掉,會是有點麻煩的一件事,不如直接設立起終點,用while迴圈把事情做完。
Spiral 的走法的固定pattern就是:向右走,向下走,向左走,向上走。
所以每個while迴圈中,只要這四個方向還有element可以走,就要走。
用四個變數rowBegin, rowEnd, colBegin, colEnd當作指針,begin從0開始,end從最後一個index開始,
每走完一圈要往內圈走時,begin++,end--。
直到begin>end,代表已無路可走。
public class Solution {
public List<Integer> spiralOrder(int[][] matrix) {
List<Integer> res = new ArrayList<Integer>();
if (matrix.length == 0) {
return res;
}
int rowBegin = 0;
int rowEnd = matrix.length-1;
int colBegin = 0;
int colEnd = matrix[0].length - 1;
while (rowBegin <= rowEnd && colBegin <= colEnd) {
// Traverse Right
for (int j = colBegin; j <= colEnd; j ++) {
res.add(matrix[rowBegin][j]);
}
rowBegin++;
// Traverse Down
for (int j = rowBegin; j <= rowEnd; j ++) {
res.add(matrix[j][colEnd]);
}
colEnd--;
if (rowBegin <= rowEnd) {
// Traverse Left
for (int j = colEnd; j >= colBegin; j --) {
res.add(matrix[rowEnd][j]);
}
}
rowEnd--;
if (colBegin <= colEnd) {
// Traver Up
for (int j = rowEnd; j >= rowBegin; j --) {
res.add(matrix[j][colBegin]);
}
}
colBegin ++;
}
return res
}
}