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

tags: LeetCode C++