leetcode 54 螺旋矩陣

### 題目

給定一個包含 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]


### 思路

        方法有點取巧了,首先建立一個m*n的輔助矩陣用來表示該位置是否遍歷過了,然後設置四個方向矩陣,最開始向右,然後向下,然後向左,最後向上,這樣四個一次循環,直到將所有的元素都遍歷一遍就結束了,具體可以看代碼註釋。

### code

    

class Solution {
    public:
        vector<int> spiralOrder(vector<vector<int>>& matrix) {
            int l1=matrix.size();
            if(l1==0)
                return {};
            int l2=matrix[0].size();
            int num=1;
            vector<int>res;
            vector<vector<bool>>flag(l1,vector<bool>(l2,true));//輔助矩陣
            int turn=0;
            int i=0,j=0;
            vector<vector<int>>fangxiang;//四個方向矩陣
            fangxiang.push_back({0,1});
            fangxiang.push_back({1,0});
            fangxiang.push_back({0,-1});
            fangxiang.push_back({-1,0});
            res.push_back(matrix[i][j]);//先將(0,0)放入res中
            flag[0][0]=false;
            while(num!=l1*l2)
            {   turn=turn%4;//四個方向內循環
                if(i+fangxiang[turn][0]>=0&&i+fangxiang[turn][0]<l1&&j+fangxiang[turn][1]>=0&&j+fangxiang[turn][1]<l2&&flag[i+fangxiang[turn][0]][j+fangxiang[turn][1]]==true)//是否越界以及是否遍歷過了
                {
                    i+=fangxiang[turn][0];
                    j+=fangxiang[turn][1];
                     res.push_back(matrix[i][j]);
                    flag[i][j]=false;
                    num++;
                }
                else//轉向
                {
                    turn++;
                }
            }
            return res;
        }
    };

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章