LeetCode[Array]: Spiral Matrix II

来源:互联网 发布:js 在div写代码怎么写 编辑:程序博客网 时间:2024/05/14 04:49

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;}


0 0