本題與No.54 LeetCode題目 “螺旋矩陣”相同。
題目描述
給定一個包含 m x n
個元素的矩陣(m 行, n 列),請按照順時針螺旋順序,返回矩陣中的所有元素。
示例 1:
輸入:
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
輸出: [1,2,3,6,9,8,7,4,5]
示例 2:
輸入:
[
[1, 2, 3, 4],
[5, 6, 7, 8],
[9,10,11,12]
]
輸出: [1,2,3,4,8,12,11,10,9,5,6,7]
解題思路
本題是考察二維數組的一道題。
爲二維數組分爲設置四個邊界:left, right, top, bottom。
當left > right
或 top > bottom
時循環結束,否則按照“從左到右”→“從上到下”→“從右到左”→“從下到上”的順序掃描二維數組,如下圖所示:
每掃描完一行或者一列後需要重新定義邊界,並判斷是否結束:
具體實現細節請參考代碼及註釋。
完整代碼
class Solution {
public:
vector<int> spiralOrder(vector<vector<int>>& matrix) {
//如果二維數組爲空
if(matrix.empty() == 1){
return {};
}
vector<int> ans;
//m行n列的二維數組
int m = matrix.size();
int n = matrix[0].size();
//利用上下左右四個邊界來解決問題
int left = 0;
int right = n - 1;
int top = 0;
int bottom = m - 1;
while(true){
//從左到右掃描
for(int i = left;i<=right;i++){
ans.push_back(matrix[top][i]);
}
top++;
if(top>bottom)break;
//從上到下掃描
for(int i = top;i<=bottom;i++){
ans.push_back(matrix[i][right]);
}
right--;
if(left>right)break;
//從右到左掃描
for(int i = right;i>=left;i--){
ans.push_back(matrix[bottom][i]);
}
bottom--;
if(top>bottom)break;
//從下到上掃描
for(int i = bottom;i>=top;i--){
ans.push_back(matrix[i][left]);
}
left++;
if(left>right)break;
}
return ans;
}
};