首页 > 代码库 > LeetCode[Array]: Spiral Matrix II

LeetCode[Array]: Spiral Matrix II

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

这个题目跟LeetCode[Array]: Spiral Matrix不同的是:这个题目并不适合用递归,因为vector<vector<int> >不好递归生成。

可以按照其迭代思路来做,不过这里需要注意的一个技术问题是vector<vector<int> > matrix的初始化,一开始我是按照这个方法进行初始化的:

    vector<vector<int> > matrix(n);
    for (int i = 0; i < n; ++i) {
        vector<int> row(n, 0);
        matrix[i] = row;
    }


后来发现C++允许这样完成vector的初始化:vector<vector<int> > matrix(n, vector<int> (n, 0));

我的C++代码实现如下:

vector<vector<int> > generateMatrix(int n) {
    vector<vector<int> > matrix(n, vector<int> (n, 0));

    int beginRow = 0, endRow = n - 1, beginCol = 0, endCol = n - 1;
    int element = 1;
    while (beginRow <= endRow && beginCol <= endCol) {
        for (int j = beginCol; j <= endCol; ++j)
            matrix[beginRow][j] = (element++);
        if (++beginRow > endRow) break;

        for (int i = beginRow; i <= endRow; ++i)
            matrix[i][endCol] = (element++);
        if (--endCol < beginCol) break;

        for (int j = endCol; j >= beginCol; --j)
            matrix[endRow][j] = (element++);
        if (--endRow < beginRow) break;

        for (int i = endRow; i >= beginRow; --i)
            matrix[i][beginCol] = (element++);
        if (++beginCol > endCol) break;
    }

    return matrix;
}


LeetCode[Array]: Spiral Matrix II