3.26刷题（矩阵模拟专题）

1.59. 螺旋矩阵 II - 力扣（LeetCode）

//方法一：变换方向法
class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> dirct = {{0, 1}, {1, 0}, {0, -1}, {-1, 0}};
        vector<vector<int>> ans(n,vector<int>(n,0));
        int i=0,j=0;
        int flag=0;
        //进行两次计算，第一次为下一步的预测位置，用来判断是否溢出，用来改变flag
        //第二次计算为下一步应该填入的位置
        for(int val=1;val<=n*n;val++){
            ans[i][j]=val;
            int temp_i=i+dirct[flag][0];
            int temp_j=j+dirct[flag][1];
            if(temp_i==n||temp_j==n||temp_i<0||temp_j<0||ans[temp_i][temp_j]){
                flag=(flag+1)%4;
            }
            i+=dirct[flag][0];
            j+=dirct[flag][1];
        }
        return ans;
    }
};

2.LCR 146. 螺旋遍历二维数组 - 力扣（LeetCode）

class Solution {
public:
    vector<int> spiralArray(vector<vector<int>>& array) {
        if(array.empty()){
            return {};
        }
        vector<vector<int>> dir={{0,1},{1,0},{0,-1},{-1,0}};
        int m = array.size(), n = array[0].size();
        vector<int> ans(m * n);
        int i=0,j=0,flag=0;
        for(int k=0;k<m*n;k++){
            ans[k]=array[i][j];
            array[i][j]=INT_MAX;
            int temp_i=i+dir[flag][0];
            int temp_j=j+dir[flag][1];
            if(temp_i<0||temp_i==m||temp_j<0||temp_j==n||array[temp_i][temp_j]==INT_MAX){
                flag=(flag+1)%4;
            }
            i += dir[flag][0];
            j += dir[flag][1];
        }
        return ans;
    }
};

class Solution {
public:
    vector<int> spiralArray(vector<vector<int>>& array) {
        if (array.empty()) {
            return {};
        }

        //1，初始化变量
        vector<vector<int>>dirct={{0,1},{1,0},{0,-1},{-1,0}};
        int flag=0;
        int m=array.size(),n=array[0].size();
        int size=m*n;
        vector<int>ans;
        int x=0,y=-1;

        //2.开始模拟
        while(ans.size()<size){
            for(int k=0;k<n;k++){
                x+=dirct[flag][0];
                y+=dirct[flag][1];
                ans.push_back(array[x][y]);
            }
            m--;
            swap(m,n);
            flag=(flag+1)%4;
        }

        //3.结果输出
        return ans;

    }
};