本题并不涉及到什么算法,就是模拟过程,按照左到右、上到下、右到左、下到上 的顺序由外向内一圈一圈画下去,要注意边界情况,是否溢出以及当前位置是否已经访问过。
cpp
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
//0为未填充,非0则已填充
vector<vector<int>> matrix(n, vector<int>(n, 0));
int line = 0, column = 0;
for(int i = 1; i <= n * n;){
//从左到右
while(column < n && matrix[line][column] == 0){
matrix[line][column] = i++;
column++;
}
column--;
line++;
//从上到下
while(line < n && matrix[line][column] == 0){
matrix[line][column] = i++;
line++;
}
line--;
column--;
//从右到左
while(column >= 0 && matrix[line][column] == 0){
matrix[line][column] = i++;
column--;
}
column++;
line--;
//从下到上
while(line >= 0 && matrix[line][column] == 0){
matrix[line][column] = i++;
line--;
}
line++;
column++;
}
return matrix;
}
};
同样类型的题目:54.螺旋矩阵
同样的思路,但由于题目中没有具体给出几行几列,要知道怎么得到vector二维数组的行数和列数。
cpp
// 行数 // 列数
int rows = matrix.size(), columns = matrix[0].size();
int total = rows * columns;