螺旋矩阵 II
给你一个正整数?n ?,生成一个包含?1 ?到?n2 ?所有元素,且元素按顺时针顺序螺旋排列的?n x n ?正方形矩阵?matrix ?。
示例 1:
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
示例 2:
输入:n = 1
输出:[[1]]
提示:
以下程序实现了这一功能:
#include <bits/stdc++.h>
using namespace std;
class Solution
{
public:
vector<vector<int>> generateMatrix(int n)
{
vector<vector<int>> ans(n, vector<int>(n, 0));
int i, j = 0, time = 0, cnt = 1; //time记录第几圈
ans[0][0] = 1;
while (cnt < n * n)
{
for (i = time, j++; j < n - time && cnt < n * n; j++)
ans[i][j] = ++cnt;
for (j--, i++; i < n - time && cnt < n * n; i++)
ans[i][j] = ++cnt;
for (i--, j--; j >= time && cnt < n * n; j--)
ans[i][j] = ++cnt;
for (j++, i--; i > time && cnt < n * n; i--)
ans[i][j] = ++cnt;
time++;
}
return ans;
}
};
|