[59] Spiral Matrix II

1. 题目描述

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 ]
]

题目的意思是，给定一个数字n，输出一个以螺旋方式填满的n*n的二维数组

2. 解题思路

采用螺旋排列时的顺序为，右->下->左->上，都是走到头后换下一个方向，直到四个方向都无法前进则数组填充完成。

3. Code

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector <int> > matrix(n ,vector<int>(n,0));    //n*n的二维vector，所有元素为0
        if(n == 0)
            return matrix;
        // 右下左上
        bool flag = true;  // 标记当前填充是否完成
        string des[4] = {"right", "down", "left", "up"}; // right->down->left->up
        int choose = 0;  // 当前方向
        string curDes = des[choose%4];
        int i(0), j(0);  // i->行 j->列
        int count(1);
        matrix[0][0] = 1;
        if(n == 1)
            return matrix;
        while(flag)
        {
            int num(0);
            while(curDes == "right")
            {
                // 没有越界且右边是0
                if(j < n-1 && (matrix[i][j+1] == 0))
                {
                    matrix[i][++j] = ++count;
                    num++;
                }
                else
                {
                    curDes = des[++choose%4];  // 换方向
                }
                if(num == 0)
                    return matrix;
            }
            while(curDes == "down")
            {
                // 没有越界且下边是0
                if(i < n-1 && (matrix[i+1][j] == 0))
                {
                    matrix[++i][j] = ++count;
                }
                else
                {
                    curDes = des[++choose%4];  // 换方向
                }
            }
            while(curDes == "left")
            {
                // 没有越界且左边是0
                if(j > 0 && (matrix[i][j-1] == 0))
                {
                    matrix[i][--j] = ++count;
                }
                else
                {
                    curDes = des[++choose%4];  // 换方向
                }
            }
            while(curDes == "up")
            {
                // 没有越界且上边是0
                if(i > 0 && (matrix[i-1][j] == 0))
                {
                    matrix[--i][j] = ++count;
                }
                else
                {
                    curDes = des[++choose%4];  // 换方向
                }
            }
        }
    }
};

4. 学到的知识

学到了一个创建并初始化m*n的vector二维数组的方式，使用vector的构造函数matrix(int num, int value)

//m*n的二维vector，所有元素为0
vector<vector <int> > matrix(m ,vector<int>(n,0));