《算法零基础100讲》(第3讲) 矩阵练习

最新推荐文章于 2022-08-21 11:27:10 发布

孤独的饺子

最新推荐文章于 2022-08-21 11:27:10 发布

阅读量750

点赞数 1

文章标签：矩阵线性代数 c++

本文链接：https://blog.csdn.net/qq_43848497/article/details/121469430

版权

数据结构与算法分析专栏收录该内容

3 篇文章 0 订阅

订阅专栏

力扣例题：螺旋矩阵

思路描述：简单模拟，依次往右下左右走到尽头并且循环，尽头可以用判断数组边界+标记数组（用来标记该单元格是否走过）的方法来实现，显然，该方法一定可行并且总的步数可以预知（步数为矩阵的元素数目），所以，我们可以通过判断步数是否达到总步数判断结束条(也可以通过判断无路可走来实现）。

代码如下：

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        int m = matrix.size();
        int n = matrix[0].size();
        int cnt = m*n;
        int x_label = 0,y_label = 0;
        int iterator = 1;
        int i = 0;
        vector<vector<int>> dir(4,vector<int>(2,0));
        vector<vector<int>> book(m,vector<int>(n,0));
        vector<int> ans(cnt,0);
        ans[0] = matrix[0][0];
        book[0][0] = 1;
        //标记数组，判断是否走过
        dir[0][0] = 0,dir[0][1] = 1;
        dir[1][0] = 1,dir[1][1] = 0;
        dir[2][0] = 0,dir[2][1] = -1;
        dir[3][0] = -1,dir[3][1] = 0;
        //初始化四个方向数组
        while(true){
            if(x_label+dir[i][0]<=m-1 && x_label+dir[i][0]>=0
            && y_label+dir[i][1]>=0 && y_label+dir[i][1]<=n-1
            && book[x_label+dir[i][0]][y_label+dir[i][1]]==0){
                x_label = x_label + dir[i][0];
                y_label = y_label + dir[i][1];
                book[x_label][y_label] = 1;
                ans[iterator++] = matrix[x_label][y_label];
            }
            else{
                i = (i+1)%4;
            }
            if(iterator == cnt) break;
        }
        return ans;
    }
};