Lintcode 381. 螺旋矩阵 II 题解

描述

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

样例

n = 3
矩阵为

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

思路：和上一题打印螺旋数组思路相同，只不过这次改成了往数组里面填数字。

class Solution {
public:
    /**
     * @param n: An integer
     * @return: a square matrix
     */
    vector<vector<int>> generateMatrix(int n) {
        // write your code here
        vector<vector<int>> matrix(n,vector<int>(n,0));
        int xlen = n;
        int ylen = n;
        int direct = 0;
        int index_x = 0;
        int index_y = -1;//这个入口坐标还是挺重要的，每次都需要确定一条边遍历的起始位置！
        int num = 1;
        while(xlen && ylen){//循环出口
            if(direct == 0){
                for(int i = 1;i <= xlen; ++i){
                    matrix[index_x][index_y+i] = num++;
                }
                index_y += xlen;
                ylen--;
            }else if(direct == 1){
                for(int i = 1;i <= ylen;++i){
                    matrix[index_x+i][index_y] = num++;
                }
                index_x += ylen;
                xlen--;
            }else if(direct == 2){
                for(int i = 1;i <= xlen; ++i){
                    matrix[index_x][index_y-i] = num++;
                }
                index_y -= xlen;
                ylen--;
            }else{
                for(int i = 1;i <= ylen;++i){
                    matrix[index_x-i][index_y] = num++;
                }
                index_x -= ylen; 
                xlen--;
            }
            direct = (direct+1)%4;//每次改变方向，0向右，1下，2左，3上。
        }
        return matrix;
    }
};