天天看點

PAT甲級 1105 Spiral Matrix (25分)

題目連結:

https://pintia.cn/problem-sets/994805342720868352/problems/994805363117768704

思路:

1.将所給序列倒序排序;

2.一圈一圈地将序列逆時針填入矩陣;我的思路是使用變量x,y來表示目前我們所需操作的矩陣位置,然後按如下順序挨個填入資料;

PAT甲級 1105 Spiral Matrix (25分)

重點是對x,y坐标位置的控制;

代碼:

#include <bits/stdc++.h>

using namespace std;

int N, m, n;

int main() {
#ifdef MyTest
	freopen("Sakura.txt", "r", stdin);
#endif
	scanf("%d", &N);
	vector<int> arr(N);
	for(int i = 0; i < N; ++i) scanf("%d", &arr[i]);
	sort(arr.begin(), arr.end(), greater<int>());
	n = sqrt(N);
	while(N % n) --n;
	m = N / n;
	vector<vector<int> > mat(m, vector<int>(n));
	int mm = m, nn = n, pos = 0, x = 0, y = 0;
	while(pos < N) {
		for(int i = 0; i < nn; ++i) mat[x][y++] = arr[pos++];
		--y, ++x;
		if(pos >= N) break;
		for(int i = 0; i < mm - 2; ++i) mat[x++][y] = arr[pos++];
		if(pos >= N) break;
		for(int i = 0; i < nn; ++i) mat[x][y--] = arr[pos++];
		if(pos >= N) break;
		++y, --x;
		for(int i = 0; i < mm - 2; ++i) mat[x--][y] = arr[pos++];
		++x, ++y;
		mm -= 2;
		nn -= 2;
	}
	for(int i = 0; i < m; ++i) {
		for(int j = 0; j < n; ++j) {
			if(j) putchar(' ');
			printf("%d", mat[i][j]);
		}
		putchar('\n');
	}
	return 0;	
}