首页 > 代码库 > 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
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。