题目描述
示例 1:
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
示例 2:
输入:n = 1
输出:[[1]]
生成螺旋顺序的数组和遍历螺旋顺时针的数组的方法是相同的,不同的是,上一种是访问对应的值,该方法是将对应的位置,按照顺序进行存储
1.首先使用flag标识是否访问过该元素
2.ret保存输出的值
3.这里遍历的方式,首先按照一个方向遍历,如果当前方向已经遍历完成,则换一个方向遍历,当遍历完成时,那么结束
注:这里有所不同,这里加不加flag矩阵都是可以的,因为这里要返回ret矩阵,若是对当前位置已经填过值了,那么直接就不填,所以他和flag元素有相同的作用
这里时间复杂度是O(n2),空间复杂度是O(1)
vector<vector<int>> generateMatrix(int n) {
//vector<vector<int>> flag(n, vector<int>(n));
vector<vector<int>> ret(n, vector<int>(n));
static int dirctions[4][2] = { { 0, 1 }, { 1, 0 }, { 0, -1 }, { -1, 0 } };
int dirc_num = 0;
int i = 0, j = 0;
for(int k = 1; k <= n * n; k++) {
ret[i][j] = k;
//flag[i][j] = 1;
int n_i = i + dirctions[dirc_num][0];
int n_j = j + dirctions[dirc_num][1];
if(n_i >= 0 && n_i < n && n_j >= 0 && n_j < n && !ret[n_i][n_j]) { //如果下一个位置合法
i = n_i;
j = n_j;
} else { //如果不合法
dirc_num = (dirc_num + 1) % 4 ;
i += dirctions[dirc_num][0];
j += dirctions[dirc_num][1];
}
}
return ret;
}
按层输出
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> ret(n, vector<int>(n));
int ind = 1;
int top = 0, bottom = n - 1, left = 0, right = n - 1; //定义四个边界
while(ind <= n * n) {
//遍历上
for(int i = left; i <= right; i++) {
ret[top][i] = ind++;
}
//遍历右
for(int i = top + 1; i <= bottom; i++) {
ret[i][right] = ind++;
}
//只有多行且多列才需要下遍历和左遍历
if(bottom > top && right > left) {
//遍历下
for(int i = right - 1; i >= left; i--) {
ret[bottom][i] = ind++;
}
//遍历左
for(int i = bottom - 1; i >= top + 1; i--) {
ret[i][left] = ind++;
}
}
top++, left++, bottom--, right--;
}
return ret;
}
int main() {
generateMatrix(3);
return 0;
}