LeetCode-59-Spiral Matrix II

最新推荐文章于 2021-05-17 02:05:37 发布

weixin_33824363

最新推荐文章于 2021-05-17 02:05:37 发布

阅读量97

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/nobodywang/p/10337300.html

本文介绍了一种用于生成螺旋矩阵的算法，输入一个正整数n，可以生成一个n*n的矩阵，矩阵元素从1到n^2按螺旋顺序填充。通过控制四个边界，即上界、下界、左界和右界，实现矩阵的螺旋填充。

算法描述：

Given a positive integer n, generate a square matrix filled with elements from 1 to n2 in spiral order.

Example:

Input: 3
Output:
[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]

解题思路：模拟，注意边界。

    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> results(n,vector<int>(n,0));
        int rowBegin = 0;
        int rowEnd = n-1;
        int colBegin = 0;
        int colEnd = n-1;
        int index = 1;
        while(rowBegin <= rowEnd && colBegin <=colEnd){
            for(int i=colBegin; i <= colEnd; i++) results[rowBegin][i] = index++;
            rowBegin++;
            for(int i=rowBegin; i <= rowEnd; i++) results[i][colEnd] = index++;
            colEnd--;
            if(rowBegin <= rowEnd && colBegin <= colEnd){
                for(int i=colEnd; i >= colBegin; i--) results[rowEnd][i] = index++;
                rowEnd--;
                for(int i=rowEnd; i >= rowBegin; i--) results[i][colBegin] = index++;
                colBegin++;
            }
        }
        return results;
    }