和昨天一样的思路
class Solution {
vector<vector<int>> matrix;
public:
vector<vector<int>> generateMatrix(int n) {
matrix.resize(n,vector<int>(n));
int row1=0;
int col1=0;
int row2=n-1;
int col2=n-1;
int count=1;
while(1){
for(int i=col1;i<=col2;i++) matrix[row1][i]=count++;
if(++row1>row2) break;
for(int i=row1;i<=row2;i++) matrix[i][col2]=count++;
if(--col2<col1) break;
for(int i=col2;i>=col1;i--) matrix[row2][i]=count++;
if(--row2<row1) break;
for(int i=row2;i>=row1;i--) matrix[i][col1]=count++;
if(++col1>col2) break;
}
return matrix;
}
};