c++与java代码如下:
具体要点
1.每次while循环中,分别按照螺旋的方向对 “上左下右” 的边进行操作
2.每条边里面的数字的遍历遵循:左闭右开,即 [ , ) 这样才能保证每次遍历的规律一致,不然考虑完这一层的边界条件,下一层的边界条件又不一样了,会把自己绕进去
3.考虑while的循环次数(转几圈):
对于n/2==0,即偶数,循环n/2次
对于n/2==1,即奇数,循环n/2 + 1次,特殊在于,当转了n/2圈后,还会剩余中心位置没有填充,所以while循环结束后,要对奇数加以判断并填充中心。
4.如何保证左闭右开:
使用offset进行偏移,每次while后offset++
举例:第一圈(最外圈)起始:i = startx, j = starty;
起始位置(startx,starty)闭 上:[ j ,n-offset) i不变,j进行遍历++,遍历结束此时j=n-offset
左:[ i ,n-offset) j不变,i进行遍历++,遍历结束此时i=n-offset
下:[ j ,starty) i不变,j进行遍历--
终止位置(startx,starty) 开 右:[ i ,startx) j不变,i进行遍历--
(写的可能比较绕,不如图示清楚)
c++代码如下
#include<vector>
using namespace std;
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> result(n, vector<int>(n, 0));
int startx = 0, starty = 0, offset = 1;
int count = 1;
int i = 0, j = 0;
int loop = n / 2;
while (loop--) {
//每次执行四条边的循环
//重置i,j
i = startx, j = starty;
//上
for (j = starty;j < n - offset;j++) {
result[startx][j] = count++;
}
//右
for (i = startx;i < n - offset;i++) {
result[i][j] = count++;
}
//下
for (;j > starty;j--) {
result[i][j] = count++;
}
//左
for (;i > startx;i--) {
result[i][j] = count++;
}
startx++, starty++, offset++;
}
if (n % 2 == 1) {
result[startx][starty] = count;
}
return result;
}
};
//测试debug
int main() {
Solution sol;
vector<vector<int>> result;
result = sol.generateMatrix(3);
}
java代码如下:
class Solution {
public int[][] generateMatrix(int n) {
int[][] nums = new int[n][n];
int startx = 0, starty = 0;
int offset = 1;
int loop = n / 2;
int i = 0, j = 0;
int count = 1;
while (loop > 0) {
i = startx;
j = starty;
// 顶部
// 左闭右开,所以判断循环结束时, j 不能等于 n - offset
for (j = starty; j < n - offset; j++) {
nums[startx][j] = count++;
}
// 右列
// 左闭右开,所以判断循环结束时, i 不能等于 n - offset
for (i = startx; i < n - offset; i++) {
nums[i][j] = count++;
}
// 底部
// 左闭右开,所以判断循环结束时, j != startY
for (; j > starty; j--) {
nums[i][j] = count++;
}
// 左列
// 左闭右开,所以判断循环结束时, i != startX
for (; i > startx; i--) {
nums[i][j] = count++;
}
startx++;
starty++;
offset++;
loop--;
}
if(n%2==1){
nums[startx][starty] = count;
}
return nums;
}
}
学习自代码随想录