螺旋矩阵 II

描述

给你一个数n生成一个包含1-n^2的螺旋形矩阵

样例

n = 3
矩阵为：

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

思考

1. 上下左右都有边界，按照顺序 (向右，向下，向左，向上) 一个轮回，边界不断缩小，所以设置了四个变量表示边界，数字递增直到填满

代码

//  By Lentitude


class Solution {
public:
    /**
     * @param n an integer
     * @return a square matrix
     */
    vector<vector<int>> generateMatrix(int n) {
        // Write your code here
        // 初始化 vector 二维动态数组
        vector<vector<int> > vec(n);
        for (int i = 0; i != n; ++i){
            vec[i].resize(n);
        }

        int value = 1;
        int x = 0;
        int x_left = 0;         //  起始的 X 索引位置
        int x_right = n - 1;    // 结束的 X 索引位置

        int y = 0;
        int y_top = 0;          // 起始的 Y 索引位置
        int y_bottom = n - 1;   // 结束的 Y 索引位置


        while (value <= n*n){

            // 000 -> 123
            while (x <= x_right  && vec[y_top][x] == 0){
                vec[y_top][x++] = value++;
            }

            // if (vec[(n + 1)/2 - 1] == n*n){
            //     return vec;
            // }

            y_top++;
            y = y_top;

            /**
             * 3  ->   3
             * 0       4
             * 0       5
             */ 

            while (y <= y_bottom && vec[y][x_right] == 0){
                vec[y++][x_right] = value++;
            }

            x_right--;
            x = x_right;


            /**
             * 005 -> 765
             */

            while (x >= x_left && vec[y_bottom][x] == 0){
                vec[y_bottom][x--] = value++;
            }

            y_bottom--;
            y = y_bottom;

            /**
             * 0   ->  9
             * 0       8
             * 7       7
             */ 

            while (y >= y_top && vec[y][x_left] == 0){
                vec[y--][x_left] = value++;
            }

            x_left++;
            x = x_left;
        }

        return vec;



    }
};