思路:弄清楚正方形的四个边界在旋转填充的时候的变化。
一开始在左上角的初始位置,从左-->右的变化,此时横坐标不变,即为up,纵坐标从左 -->右,
for(int j = left;j <= right;j++)
vec[up][j] = num++;
在变换完“左-->右”之后,下一次循环如图顺时针的变换,此时最上面的一行,应该向下走了一行,即up = up+1;判断终止的条件就是上界up > 下界down了。
if(++up > down) break;
其它三个方向同理,注意一开始写的时候要考虑什么时候跳出循环 :上界在变换之后大于下界(左界在变换之后大于有界...同理)。
代码:
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> vec(n,vector<int>(n,0));
int up = 0,left = 0,right = n-1,down = n-1,num = 1;
while(true){
for(int j = left;j <= right;j++)
vec[up][j] = num++;
if(++up > down) break;
for(int i = up;i <= down;i++)
vec[i][right] = num++;
if(--right < left) break;
for(int j = right;j >= left;j--)
vec[down][j] = num++;
if(--down < up) break;
for(int i = down;i >= up;i--)
vec[i][left] = num++;
if(++left > right) break;
}
return vec;
}
};