PAT-2023年冬季考试-乙级 B-4方格填数

开学要参加PAT乙级了，寒假买几套卷子刷一下题，结果就是被虐的很惨，只考了70分，这道题没写出来，个人感觉比25分压轴题还要难，那道题用几个容器基本就能得20多分了。

看一下题目：

2014 年哈佛-麻省理工数学竞赛中一道题是这样的：将正整数 1, 2, ..., 64 填入 8×8 的方格棋盘中，使得对任何 1≤i<64，i 和 i+1 都必须填在两个具有公共边的方格中。求棋盘上对角线中所填数的和的最大值。（注意：两个对角线都要考虑；64 和1 也必须填在两个具有公共边的方格中。）
这题有点难…… 幸好我们并不要求你写程序解决这个问题。
你的任务是：对任一给定的数字填充方案，判定其是否满足填充的条件，并且在所有给出的方案中，找出满足条件的、且对角线数字和最大的那个方案。

输入格式：

输入在一行中首先给出两个正整数 n（≤100） 和 m（≤20），分别为棋盘的规模（即棋盘有 n×n 个方格）和输入的方案数量。因为容易证明奇数 n 一定不存在满足条件的解，所以题目保证给出的 n 都是偶数。
随后给出 m 个填充方案，每个方案占 n 行，每行 n 个不超过 n2 的数字。同行数字间以空格分隔。

输出格式：

在一行中首先输出满足条件的、且对角线数字和最大的方案数。随后一行中按照递增序输出这些方案的编号（编号按输入的顺序从 1 到 m）。
注意每行数字间以 1 个空格分隔，行首尾不得有多余空格。

输入样例：

4 5
16 1 2 3
15 14 13 4
10 11 12 5
9 8 7 6
16 1 2 3
15 14 13 4
10 11 12 5
9 8 7 10
15 16 1 2
14 13 4 3
11 12 5 6
10 9 8 7
3 4 5 6
2 13 12 7
1 14 11 8
16 15 10 9
10 5 4 3
7 12 13 2
8 11 14 1
9 6 15 16

输出样例：

2
1 4

最难想的就是如何遍历一个二维矩阵了，分类讨论的话就写死了，角和边都得单独考虑，然后博主就在这里卡死了，再加上时间确实不够，转过去写最后一题了。之后问了一下学长，给我写了一个代码，但由于考试时间已经结束无法提交，不知道能不能通过所有的测试点。

他写了一个check函数，用res值的1、2、3三种不同的情况来记录判断的结果，最有意思的就是这个dx和dy数组，说是bfs算法里很常用的方法，去遍历一个二维矩阵中每个元素的周围几个元素，为了节省代码量，就用两个数组和一个for循环来表示了。

对角线求和也使用了一个双指针，从两头开始算，减少时间复杂度，太强了。

代码如下：

#include<iostream>
#include<vector>

using namespace std;

const int N = 105;

int n, m, g[N][N];
int dx[4] = { 0, 0, 1, -1 }, dy[4] = { 1, -1, 0, 0 };

bool check() {
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= n; j++) {
			int res = 0;
			if (g[i][j] == 1)	res |= 1;
			if (g[i][j] == n * n)	res |= 2;
			for (int k = 0; k < 4; k++) {
				int x = dx[k] + i, y = dy[k] + j;
				if (x < 1 || x > n || y > n || y < 0)
					continue;
				if (g[x][y] == g[i][j] - 1)	res |= 1;
				if (g[x][y] == g[i][j] + 1)	res |= 2;
			}
			if (res != 3)	return false;
		}
	}
	return true;
}

int func()  //对角线求和 
{
	int res = 0;
	for (int i = 1; i <= n; i++) {
		res += g[i][i] + g[i][n - i + 1];
	}
	return res;
}

int main() {
	cin >> n >> m;
	vector<int> ans;
	int cnt = 0, maxn = 0;
	for (int i = 1; i <= m; i++) 
	{
		for (int j = 1; j <= n; j++) {
			for (int k = 1; k <= n; k++) {
				cin >> g[j][k];
			}
		}
		if (check()) {
			int x = func();
			if (maxn == x) {
				ans.push_back(i);
				cnt++;
			}
			else if (maxn < x) {
				maxn = x;
				cnt = 1;
				ans.clear();
				ans.push_back(i);
			}
		}
	}
	cout << cnt << endl;
	for (int i = 0; i < ans.size() - 1; i++) {
		cout << ans[i] << " ";
	}
	cout << ans.back();
}