59. Spiral Matrix II

最新推荐文章于 2018-03-13 16:01:04 发布

Biglethz

最新推荐文章于 2018-03-13 16:01:04 发布

阅读量281

点赞数

分类专栏： LeetCode 文章标签： LeetCode

本文链接：https://blog.csdn.net/Biglethz/article/details/78046546

版权

LeetCode 专栏收录该内容

10 篇文章 0 订阅

订阅专栏

题目：

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

For example,
Given n = 3,

You should return the following matrix:
[
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ]
]

思路

规律：generateMatrix(n+1)时仅需将generateMatrix(n)的结果的每一个元素加上一个常数（2*n+1），然后按中心对称旋转矩阵元素(V[i][j]=v[n-i-1][n-j-1])；然后增加第一行和最后一列元素即可。
举例：generateMatrix（2）的结果为：
[[ 1, 2,],
[ 4, 3 ]]
generateMatrix（3）的结果为：
[
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ]
]
其中，左下角的方阵可以通过
[[ 1, 2,],
[ 4, 3 ]]
加上常数5(即2*2+1)然后中心对称旋转得到；然后增加第一行[1,2,3]和补全最后一列[[4],[5]]。

很多博客里面提到这个题目仅需要螺旋顺序将元素填入数组即可，这里提供另外一种思路，看起来很麻烦，但是在需要连续输出多个螺旋矩阵时，这种思路应该会快很多，因为generateMatrix(n+1)可以用到generateMatrix(n)的结果。(看到有人python只需要6行搞定，自我安慰一下)

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> vvec;
        if (n <= 0) return vvec;//边界条件处理
        vector < int > tem{ 1 };
        vvec.push_back(tem);
        int i = 2;
        while (i < n+1){
            reverse(vvec,i-1);//翻转矩阵和添加常数
            vector<int> tem;
            //增加第一行元素
            //C++的vector不能再前面插入
            //So,从最后一行插入，再交换至第一行
            for (int j = 0; j < i ; j++){
                tem.push_back(j + 1);
            }
            vvec.push_back(tem);
            int bias = i + 1;
            //补全最后一列元素
            for (int k = 0; k < i - 1;k++){
                vvec[k].push_back(bias++);
            }
            //交换
            auto lastVec = vvec[i - 1];
            for (int k = i - 1; k > 0; k--){
                vvec[k] = vvec[k - 1];
            }
            vvec[0] = lastVec;
            i++;
        }
        return vvec;
    }
    //辅助函数，矩阵翻转并添加常数
    void reverse(vector<vector<int>> &vvec,int n){
        auto bias = n * 2 + 1;
        //大小为奇数的矩阵，处理中心元素，避免其两次添加常数
        if (n % 2 == 1){
            vvec[n / 2][n / 2] += bias;
        }
        //处理对角线元素
        for (int i = 0; i < n / 2; i++){
            auto tem = vvec[i][i];
            vvec[i][i] = vvec[n - i - 1][n - i - 1] + bias;
            vvec[n - i - 1][n - i - 1] = tem + bias;
        }
        //处理非对角线元素
        for (int j = 1; j < n; j++){
            for (int k = 0; k < j; k++){
                auto tem = vvec[j][k] ;
                vvec[j][k] = vvec[n - j - 1][n - k - 1] + bias;
                vvec[n - 1 - j][n - k - 1] = tem + bias;
            }
        }
    }

};