本题要求将给定的 N 个正整数按非递增的顺序,填入“螺旋矩阵”。所谓“螺旋矩阵”,是指从左上角第 1 个格子开始,按顺时针螺旋方向填充。要求矩阵的规模为 m 行 n 列,满足条件:m×n 等于 N;m≥n;且 m−n 取所有可能值中的最小值。
输入格式:
输入在第 1 行中给出一个正整数 N,第 2 行给出 N 个待填充的正整数。所有数字不超过 104,相邻数字以空格分隔。
输出格式:
输出螺旋矩阵。每行 n 个数字,共 m 行。相邻数字以 1 个空格分隔,行末不得有多余空格。
输入样例:
12
37 76 20 98 76 42 53 95 60 81 58 93
输出样例:
98 95 93
42 37 81
53 20 76
58 60 76
思路:不太会做,知道肯定是顺时针从外到内遍历二维数组,但是搞不清下标的变化,所以参考了柳神的做法,按照如图所示的方法遍历二维数组:
即遍历上下边的全部元素和左右边的行数 - 2
个元素。也可以看到,图上一共围绕螺旋矩阵走了三圈,这个圈数即外层for循环的循环次数。
for (int i = 0; i < times; i++) //times就是走的圈数,每一圈代表每一层
{
...
}
接下来学习本题的核心——遍历四个边的代码。其中row表示行数,col表示列数,然后一个一个的看:
遍历上边:
先看行下标,上边是随着圈数往内走的,所以行下标就是i;
而列下标就有些复杂,因为每一行起点和终点都不一样,但是有规律可循。第一行头尾各去掉0个元素、第二行头尾各去掉1个元…可以发现每次头尾各去掉i个元素,所以每行遍历时从i开始,到col - i结束(因为数组从0~col - 1,所以实际上是在col - 1 - i结束);
//写成j < col - i也可以
for (int j = i; j <= col - 1 - i && cnt < total; j++)
v[i][j] = vec[cnt++]; //cnt是vec数组的下标索引
遍历右边:
列下标也是随着圈数往内走,不过这里的初始位置和上边的0不一样,这里是从最右侧即col - 1开始,所以列下标是col - 1 - i;
行下标是要遍历的数目,因为第一位已经被刚刚上边遍历过了,所以初始是从第二位(也就是下标1)开始;同样的,要留出最后一个位置给下边遍历,所以末尾是初始位置(row - 1)再往上一位(row - 2),又因为每遍历一圈就要往上走一个,所以末尾是row - 2 - i;
//从i + 1到row - 2 -i
for (int j = i + 1; j <= row - 2 - i && cnt < total; j++)
v[j][col - 1 - i] = vec[cnt++];
遍历下边:
行下标初始位置是最下边即row - 1,根据i往上走,所以是row - 1 - i;
这次是从右到左遍历,上边从左到右时范围是从i到col - 1 - i,这里刚好相反;
for (int j = col - 1 - i; j >= i && cnt < total; j--)
v[row - 1 - i][j] = vec[cnt++];
遍历左边:
左边是从下到上遍历,因为是列遍历所以只有行坐标动,范围和遍历右边正好相反,为row - 2 - i到i + 1;
而列坐标因为是随着圈数往内走,而初始位置是0,所以就是i;
for (int j = row - 2 - i; j >= i + 1 && cnt < total; j--)
v[j][i] = vec[cnt++];
这个难点攻破这题差不多就解决了,仅剩的难点就是如果找出满足m×n 等于 N;m≥n;且 m−n 取所有可能值中的最小值
这个条件的m和n了。其实很好找,取N的平方根,然后让它一边递减一边对N求余,找到余数为0时的除数,这两个数就满足条件了。
#include <iostream>
#include <cmath>
#include <vector>
#include <string>
#include <algorithm>
using namespace std;
bool cmp(int a, int b) { return a > b; }
int main()
{
int n;
cin >> n;
int col = (int)sqrt(1.0 * n); //取平方根
while (n % col != 0)
col--;n
int row = n / col, total = n;
vector<int> vec(total);
for (int i = 0; i < total; i++)
cin >> vec[i];
sort(vec.begin(), vec.end(), cmp); //按从大到小排序
vector<vector<int> > v;
//构造二维数组
for (int i = 0; i < row; i++)
{
vector<int> c(col);
v.push_back(c);
}
int cnt = 0;
//times是圈数
int times = row % 2 ? (row / 2) + 1 : row / 2;
//柳神提供了更好的计算方法: int times = row / 2 + row % 2;
for (int i = 0; i < times; i++)
{
for (int j = i; j <= col - 1 - i && cnt < total; j++) //上边
v[i][j] = vec[cnt++];
for (int j = i + 1; j <= row - 2 - i && cnt < total; j++) //右边
v[j][col - 1 - i] = vec[cnt++];
for (int j = col - i - 1; j >= i && cnt < total; j--) //下边
v[row - 1 - i][j] = vec[cnt++];
for (int j = row - 2 - i; j >= i + 1 && cnt < total; j--) //左边
v[j][i] = vec[cnt++];
}
for (int i = 0; i < row; i++)
{
for (int j = 0; j < col; j++)
{
cout << v[i][j];
if (j != col - 1) cout << ' ';
}
cout << endl;
}
return 0;
}