给你一个正整数 n
,生成一个包含 1
到 n2
所有元素,且元素按顺时针顺序螺旋排列的 n x n
正方形矩阵 matrix
。
思路:
- 填充上行从左到右
- 填充右列从上到下
- 填充下行从右到左
- 填充左列从下到上
采用左闭右开的原则
代码:
class Solution
{
public:
vector<vector<int>> generateMatrix(int n)
{
vector<vector<int>> res(n,vector<int>(n,0));
int startX = 0, startY = 0;
int i, j;
int mid = n / 2;
int loop = n / 2;
int count = 1;
int offset = 1;
while(loop--)
{
for(i = startX; i < n - offset; i++)
{
res[startY][i] = count++;
}
for(j = startY; j < n - offset; j++)
{
res[j][i] = count++;
}
for( ; i > startX; i--)
{
res[j][i] = count++;
}
for( ; j > startY; j--)
{
res[j][i] = count++;
}
startX++;
startY++;
offset++;
}
if(n % 2 == 1)
{
res[mid][mid] = count;
}
return res;
}
};