给定一个正整数 n,生成一个包含 1 到 n^2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3 输出: [ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ] ]
求解本题依然是要坚持循环不变量原则。
模拟顺时针画矩阵的过程:
-
填充上行从左到右
-
填充右列从上到下
-
填充下行从右到左
-
填充左列从下到上
由外向内一圈一圈这么画下去。 这里一圈下来,我们要画每四条边,这四条边怎么画,每画一条边都要坚持一致的左闭右开,或者左开右闭的原则,这样这一圈才能按照统一的规则画下来。
C++代码如下:
#include<iostream>
#include<vector>
#include <algorithm>
using namespace::std;
//给定一个正整数 n,生成一个包含 1 到 n ^ 2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
//
//示例 :
//
//输入: 3 输出 : [[ 1, 2, 3 ], [8, 9, 4], [7, 6, 5]]
class Solution {
public:
vector<vector<int>> generateMatrix(int n)
{
vector<vector<int>> res(n, vector<int>(n, 0)); // 使用vector定义一个二维数组
int startx = 0, starty = 0;//起始点
int loop = n / 2;//圈数
int mid = n / 2;//如果n为奇数时,单独填充
int count = 1;//用来填充数字
int offset = 1;//用来表示缩进的空格数
while (loop--)//横行x竖列y
{
int i = startx;//0行数
int j = starty;//0列数
for (j=starty;j < n-offset; j++)//第一行
{
res[startx][j] = count++;
}
for ( i = startx; i < n - offset; i++)//第一列
{
res[i][j] = count++;
}
for (;j > starty; j--)
{
res[i][j] = count++;
}
for (; i > startx; i--)
{
res[i][j] = count++;
}
// 第二圈开始的时候,起始位置要各自加1, 例如:第一圈起始位置是(0, 0),第二圈起始位置是(1, 1)
startx++;
starty++;
offset++;// offset 控制每一圈里每一条边遍历的长度
// 如果n为奇数的话,需要单独给矩阵最中间的位置赋值
if (n % 2) {
res[mid][mid] = count;
}
return res;
}
}
};
int main()
{
vector<int> v;
int num;
cout << "请输入一个正整数:" << endl;
cin >> num;
Solution solution;
solution.generateMatrix(num);
return 0;
}
-
时间复杂度 O(n^2): 模拟遍历二维矩阵的时间
-
空间复杂度 O(1)