Leetcode刷题：螺旋矩阵

最新推荐文章于 2024-12-03 14:01:50 发布

原创最新推荐文章于 2024-12-03 14:01:50 发布 · 214 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #c++

c++ 专栏收录该内容

5 篇文章

订阅专栏

该博客主要探讨了三种不同版本的螺旋矩阵问题，包括最简单的螺旋矩阵、螺旋矩阵II和螺旋矩阵III。这些矩阵的生成算法通过遍历和标记二维数组实现，有效地按螺旋顺序填充数值。文章详细展示了C++实现，并强调了在处理边界条件和空矩阵时的注意事项。

并不涉及什么算法

文章目录

最简单的螺旋矩阵

54. 螺旋矩阵
 剑指 Offer 29. 顺时针打印矩阵
以上两个题目基本相同，但29的测试用例中包含了空矩阵，所以还是要先考虑这种情况。如果矩阵为空，就不能用matrix[0].size()获取矩阵列数。
在这里插入图片描述

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {

        int m = matrix.size();
        if (m == 0) return {};
        int n = matrix[0].size();
        vector<vector<bool>> flag(m, vector<bool>(n));
        vector<int> ans(m * n);
        int x = 0, y = 0, i = 0;
        
        while (i < (m * n)) {
            
            while (y < n && !flag[x][y]) {
                ans[i] = matrix[x][y];
                flag[x][y] = true;
                i++, y++;
            }
            y--, x++;

            while (x < m && !flag[x][y]) {
                ans[i] = matrix[x][y];
                flag[x][y] = true;
                i++, x++;
            }
            x--, y--;

            while (y >= 0 && !flag[x][y]) {
                ans[i] = matrix[x][y];
                flag[x][y] = true;
                i++, y--;
            }
            y++, x--;

            while (x >= 0 && !flag[x][y]) {
                ans[i] = matrix[x][y];
                flag[x][y] = true;
                i++, x--;
            }
            x++, y++;

        }
        return ans;
    }
};

螺旋矩阵 II

59. 螺旋矩阵 II
在这里插入图片描述

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        int h = 0, v = 0, x = 0;
        vector<vector<bool>> flag(n, vector<bool>(n));
        vector<vector<int>> ans(n, vector<int>(n));
        vector<int> num;
        for (int i = 1; i <= n * n; i++) {
            num.push_back(i);
            cout << i;
        }
        while (x < n * n) {
            while (v < n && !flag[h][v]) {
                ans[h][v] = num[x];
                flag[h][v] = true;
                x++, v++;
            }
            v--, h++;
            while (h < n && !flag[h][v]) {
                ans[h][v] = num[x];
                flag[h][v] = true;
                x++, h++;
            }
            h--, v--;
            while (v >= 0 && !flag[h][v]) {
                ans[h][v] = num[x];
                flag[h][v] = true;
                x++, v--;
            }
            v++, h--;
            while (h >= 0 && !flag[h][v]) {
                ans[h][v] = num[x];
                flag[h][v] = true;
                x++, h--;
            }
            h++, v++;
        }
        return ans;

    }
};

螺旋矩阵 III

在这里插入图片描述

class Solution {
public:
    vector<vector<int>> spiralMatrixIII(int rows, int cols, int rStart, int cStart) {
        int step  = 1, cnt = 0;
        vector<vector<int>> ans(rows * cols, vector<int>(1, 2));
        vector<vector<bool>> flag(rows, vector<bool>(cols));
        while (cnt < rows * cols) {
            for (int j = 0; j < step; j++) {
                if (rStart >= 0 && rStart < rows && cStart >= 0 && cStart < cols) {
                    ans[cnt] = {rStart, cStart};
                    flag[rStart][cStart] = true;
                    cnt++;
                }
                cStart++;
            }

            for (int j = 0; j < step; j++) {
                if (rStart >= 0 && rStart < rows && cStart >= 0 && cStart < cols) {
                    ans[cnt] = {rStart, cStart};
                    flag[rStart][cStart] = true;
                    cnt++;
                }
                rStart++;
            }
            step++;

            for (int j = 0; j < step; j++) {
                if (rStart >= 0 && rStart < rows && cStart >= 0 && cStart < cols) {
                    ans[cnt] = {rStart, cStart};
                    flag[rStart][cStart] = true;
                    cnt++;
                }
                cStart--;
            }
            for (int j = 0; j < step; j++) {
                if (rStart >= 0 && rStart < rows && cStart >= 0 && cStart < cols) {
                    ans[cnt] = {rStart, cStart};
                    flag[rStart][cStart] = true;
                    cnt++;
                }
                rStart--;
            }
            step++;
        }
        return ans;
    }
};