题目: 螺旋矩阵
思路: 主要解决两个问题,一是求矩阵的行数和列数,二是按照螺旋路径填入非增序的序列。
第一个问题可以直接暴力求解,复杂度为O(104)。
对于第二个问题,我是对螺旋路径进行模拟来解决的,设定初始位置为数组左上角,dx,dy数组控制前进方向,(这里的方向顺序有多种选择,但要保证是顺时针并且第一次是向右移动的),按照上述顺序选择下一个位置(在数组范围内并且未经过)。
PS: 下一个位置的选择要从上一个位置的方向开始,否则会模拟失败。
CODE ⇩⇩⇩
#include <bits/stdc++.h>
using namespace std;
int dx[4] = {0,1,0,-1},dy[4] = {1,0,-1,0};
int lis[10005];
int main() {
int N,m{},n{};
scanf("%d",&N);
for(int i = 0;i < N;i++) scanf("%d",&lis[i]);
sort(lis,lis+N,greater<int>());
for(n = (int)sqrt(N);n >= 1 && (N % n);n--);
m = N / n;
int matrix[m][n];
memset(matrix,0,sizeof(matrix));
int x = 0,y = 0,k = 0,pos = 0;
while(k < N){
matrix[x][y] = lis[k++];
for(int i = pos;(i+1) % 4 != pos;i = (i+1) % 4){
int nx = x + dx[i],ny = y + dy[i];
if((0 <= nx && nx < m) && (0 <= ny && ny < n) && !matrix[nx][ny]){
pos = i,x = nx,y = ny;
break;
}
}
}
for(int i = 0;i < m;i++){
for(int j = 0;j < n;j++){
printf("%d%s",matrix[i][j],j == n - 1 ? "\n" : " ");
}
}
return 0;
}