题目:是Spiral Matrix相关的的。这题的意思是给定一个n,那么在n*n的矩阵里按照循环记录将1,2,3,..., n*n。如下如果给定3,那么:
[ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ] ]一开始我想是不是有数学公式直接下标对应的,那直接遍历输出就可以了。但是推了一会,没有什么好的头绪。于是就分情况讨论了,和Spiral Matrix一样的分情况,只是这里的分情况是记录在数组里面。利用上,下,左,右,分别记录每次循环可以到达的四个边界,我们每次就输出一环,走完一环往内走。需要注意的:1.要用下标访问vector<vector> >需要先声明大小才可以使用下标访问。2.每走完一环,相应更新四个方向。3.如果是奇数,那么n*n这个数需要另外存。也就是存在[n/2][n/2]的位置。
class Solution { public: vector<vector<int> > generateMatrix(int n) { vector<int> sub(n); vector<vector<int> > ans(n, sub); // 也可以用vector<vector<int> > ans(n, vector<int>(n)); if (n < 1) return ans; int val = 1, left = 0, right = n-1, up = 1, down = n-1; for (int i = 0; i < n/2; i++) { for (int j = left; j <= right; j++) { ans[i][j] = val++; } for (int j = up; j <= down; j++) { ans[j][right] = val++; } for (int j = right - 1; j>=left; j--) { ans[down][j] = val++; } for (int j = down - 1; j >= up; j--) { ans[j][left] = val++; } left++; right--; up++; down--; } if (n%2) ans[n/2][n/2] = n*n; return ans; } };
时间: 2024-10-05 12:49:52