leetcode面试经典150题——35 螺旋矩阵

题目: 螺旋矩阵

描述
给你一个 m 行 n 列的矩阵 matrix ,请按照 顺时针螺旋顺序 ,返回矩阵中的所有元素。

示例:
leetcode面试经典150题——35 螺旋矩阵_第1张图片
输入:matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出:[1,2,3,6,9,8,7,4,5]

提示:

m == matrix.length
n == matrix[i].length
1 <= m, n <= 10
-100 <= matrix[i][j] <= 100
leetcode链接

方法一:模拟
我们顺时针螺旋输出矩阵的顺序可以看成为输出一个个子矩阵的边框,即每个矩阵的第一行,最后一列,最后一行,第一列,然后去除掉重复输出的四个对角,再对下一个子矩阵循环此输出,考虑特殊情况当子矩阵为一行或者一列时候,直接输出一行或者一列。

如下图所示,螺旋输出该矩阵
leetcode面试经典150题——35 螺旋矩阵_第2张图片
第一次依次输出最外层矩阵四个边框,然后去除重复元素(即4个角多输出了一次)
leetcode面试经典150题——35 螺旋矩阵_第3张图片
再输出子矩阵的四个边框,重复操作,直到没有子矩阵
leetcode面试经典150题——35 螺旋矩阵_第4张图片
leetcode面试经典150题——35 螺旋矩阵_第5张图片

时间复杂度:o(mn) 矩阵每个元素遍历一次,一共m*n个元素
空间复杂度:o(1)

vector<int> spiralOrder(vector<vector<int>>& matrix) {
        vector<int> ans;
        int m  = matrix.size(),n = matrix[0].size();
        int r = 0,l = 0;
        while(m>0&&n>0){
            for(int j=l;j<l+n;j++){//输出第一行
                ans.push_back(matrix[r][j]);
            }
            if(m==1)break;
            ans.pop_back();//取出重复存储的元素
            for(int i=r;i<r+m;i++){//输出最后一列
                ans.push_back(matrix[i][l+n-1]);
            }
            if(n==1)break;
            ans.pop_back();//取出重复存储的元素
            for(int j=l+n-1;j>=l;j--){//输出最后一行
                ans.push_back(matrix[r+m-1][j]);
            }
            ans.pop_back();//取出重复存储的元素
            for(int i=r+m-1;i>=r;i--){//输出第一列
                ans.push_back(matrix[i][l]);
            }
            ans.pop_back();//取出重复存储的元素
            m-=2,n-=2;//更新下一个子矩阵的行列数
            r++,l++;//更新下一个子矩阵的第一个元素的行列下标
        }
        return ans;
    }

你可能感兴趣的:(leetcode,面试,矩阵)