剑指 Offer | 29. 顺时针打印矩阵

题目:

输入一个矩阵,按照从外向里以顺时针的顺序依次打印出每一个数字。

示例 1:

输入:matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出:[1,2,3,6,9,8,7,4,5]

示例 2:

输入:matrix = [[1,2,3,4],[5,6,7,8],[9,10,11,12]]
输出:[1,2,3,4,8,12,11,10,9,5,6,7]

限制:

0 <= matrix.length <= 100
0 <= matrix[i].length <= 100

思路:

  • 一圈一圈循环打印。
class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        if(matrix.empty() || matrix[0].size() == 0) return vector<int>();
        int columns = matrix[0].size(), rows = matrix.size();
        vector<int> res;
        int pivot = 0; // 每个圈的起点
        while(columns > pivot * 2 && rows > pivot * 2)
        {
            // 循环调用 loop 函数来输入。循环终止条件解释在上文中。
            loop(matrix, columns, rows, pivot, res);
            pivot ++;
        }
        return res;
    }
    void loop(vector<vector<int>>& matrix, int columns, int rows, int pivot, vector<int>& res)
    {
        // loop 函数用来向 res 数组中输入一整圈的元素
        int rowEnd = rows - 1 - pivot;
        int colEnd = columns - 1 - pivot;
        for(int i=pivot; i<=colEnd; ++i)
        {
            // 第一步:从左到右打印一行
            res.push_back(matrix[pivot][i]);
        }
        if(pivot < rowEnd)
        {
            for(int i=pivot+1; i<=rowEnd; ++i)
            {
                // 第二步:从上到下打印一列
                res.push_back(matrix[i][colEnd]);
            }
        }
        if(pivot < rowEnd && pivot < colEnd)
        {
            for(int i=colEnd-1; i>=pivot; --i)
            {
                // 第三步:从右到左打印一行
                res.push_back(matrix[rowEnd][i]);
            }
        }
        if(pivot <= rowEnd - 2 && pivot < colEnd)
        {
            for(int i=rowEnd-1; i>pivot; --i)
            {
                // 第四步:从下到上打印一列
                res.push_back(matrix[i][pivot]);
            }
        }
    }
};

参考:字节题库 - #剑29 - 简单 - 顺时针打印矩阵 - 3刷

上一篇:在 n 个数当中找第k小元素 (BFPRT算法,最坏情况为线性时间的选择问题)


下一篇:算法第二章上机实践报告