【代码超详解】CodeForces 1335E1. Three Blocks Palindrome (easy version)

一、传送门

https://codeforces.com/contest/1335/problem/E1

二、算法分析说明与代码编写指导

设 r 为结果，s[i] 代表数 i 的已出现次数的前缀和，变长数组 p[i] 代表数字 i 出现的所有位置。
输入的时候要将各个数字出现的位置都记下来，并统计到字符串的每个位置的累计出现次数。
构造要求的回文串有两种情况：
第一种是 y > 0，x = 0，即只有第二部分长度不为零。
第二种是 y > 0，x > 0，即三个部分的长度都不为零。
对第一种情况，先统计一遍各个数字的出现次数，这些数字全部选到第二段，就构造完毕。记录最长的长度为答案。
对第二种情况，考察每个出现次数大于 1 的数字（否则不能平均塞到头尾两段）i 。p[i][j] 和 p[i][k] 代表当前正在枚举的头尾两个位置，分别代表三段回文串的第一段末尾和第三段开头的位置。选出这两个位置之间（不含这两个位置）出现次数最高的数字，就可以计算出要求的三段回文串的长度。枚举的方向是从两端到中间。但有的博客给出了从中间到两端枚举的方案，也是可以的。

三、AC 代码

#include<cstdio>
#include<algorithm>
#include<vector>
#pragma warning(disable:4996)
using namespace std;
const unsigned nmax = 2001;
unsigned t, n, a[nmax], r, s[27][nmax]; vector<unsigned> p[27];
int main() {
	scanf("%u", &t); ++t;
	while (--t) {
		scanf("%u", &n); r = 0;
		for (unsigned i = 1; i <= 26; ++i) { p[i].clear(); fill(s[i] + 1, s[i] + 1 + n, 0); }
		for (unsigned i = 1; i <= n; ++i) {
			scanf("%u", &a[i]); p[a[i]].emplace_back(i);
			for (unsigned j = 1; j <= 26; ++j)s[j][i] = s[j][i - 1] + (j == a[i]);
		}
		for (unsigned i = 1; i <= 26; ++i) {
			r = max(r, (unsigned)p[i].size()); if (p[i].size() <= 1)continue;
			for (unsigned j = 0, k = p[i].size() - 1, J = 2; j < k; ++j, --k, J += 2) {
				for (unsigned l = 1; l <= 26; ++l) { r = max(r, J + s[l][p[i][k] - 1] - s[l][p[i][j]]); }
			}
		}
		printf("%u\n", r);
	}
	return 0;
}