HackMD
【LeetCode】 54. Spiral Matrix
Description
Given a matrix of m x n elements (m rows, n columns), return all elements of the matrix in spiral order.
給一m*n大小的矩陣(m個直排、n個橫排)，按照螺旋順序回傳裡面的元素。
Example:
Example 1:
Input:
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
Output: [1,2,3,6,9,8,7,4,5]
Example 2:
Input:
[
[1, 2, 3, 4],
[5, 6, 7, 8],
[9,10,11,12]
]
Output: [1,2,3,4,8,12,11,10,9,5,6,7]
Solution
- 用一個變數紀錄現在往哪個方向，看過的元素就加到輸出陣列並從初始矩陣中刪除。
- 需要擔心的是往下或往上跑的時候，如果有
vector因此變成空的，要將他從矩陣中移除。
Code
class Solution {
public:
vector<int> spiralOrder(vector<vector<int>>& matrix) {
vector<int> ans;
for(int dir=0;matrix.size()!=0;dir++)
{
for(int i=matrix.size()-1;i>=0;i--)
{
if(matrix[i].size()==0)
matrix.pop_back();
}
if(matrix.size()==0) break;
if(dir%4==0)
{
for(int i=0;i<matrix[0].size();i++)
{
ans.push_back(matrix[0][i]);
}
matrix.erase(matrix.begin());
}
else if(dir%4==1)
{
for(int i=0;i<matrix.size();i++)
{
ans.push_back(matrix[i].back());
matrix[i].pop_back();
}
}
else if(dir%4==2)
{
for(int i=matrix.back().size()-1;i>=0;i--)
{
ans.push_back(matrix.back()[i]);
}
matrix.pop_back();
}
else
{
for(int i=matrix.size()-1;i>=0;i--)
{
ans.push_back(matrix[i][0]);
matrix[i].erase(matrix[i].begin());
}
}
}
return ans;
}
};
