Codeforces Round #634 (Div. 3) ——E2. Three Blocks Palindrome (hard version)题解

题目链接：http://codeforces.com/contest/1335/problem/D

与easy version 不同的是，数据量变大了，所以需要改变算法。
维护二维vector记录每个数字出现的位置，并用二维数组cnt(i, j)记录节点 i 之前数字 j 出现的次数（就是easy version 里的前缀和）

改进算法是对200个可能出现的数字做循环：
（1）sz记录该数字出现的次数，并更新当前res
（2）指针 j 从第一次出现的位置到 最中间出现的位置从左向右遍历，另一个指针 sz - j - 1从右向左遍历，两个指针与左右分界中间数字 i 都出现了 j 次
（3）j 到 sz - j - 1中间，嵌套 k 的循环，记录数字k出现的次数，即前缀和做差，再更新结果 res

代码如下：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxn = 200010;
int a[maxn], cnt[maxn][210];

int main() {
    int t, n;
    cin >> t;
    while (t--) {
        vector<int> pos[210];
        cin >> n;
        for(int i = 0; i <= n; i++) {
            for(int j = 0; j <= 200; j++) {
            	cnt[i][j] = 0;
			}
        }
        for(int i = 1; i <= n; i++) {
            cin >> a[i];
            memcpy(cnt[i], cnt[i - 1], sizeof(cnt[i - 1]));		//更新 
            cnt[i][a[i]]++;
            pos[a[i]].push_back(i);		//记录位置 
        }
        int res = 0;
        for (int i = 1; i <= 200; i++) {
            int sz = pos[i].size();
            res = max(res, sz);
            for (int j = 0; j < sz / 2; j++) {
                int l = pos[i][j], r = pos[i][sz - j - 1] - 1;
                for (int k = 1; k <= 200; k++) {
                    int midsum = cnt[r][k] - cnt[l][k];	// k 出现的次数 
                    res = max(res, (j + 1) * 2 + midsum);
                }
            }
        }
        cout << res << endl;
    }
}