CodeForces - 1335 E2. Three Blocks Palindrome (hard version)
题目地址:
https://codeforces.com/contest/1335/problem/E2
基本题意:
给你一个数组,找到满足以下这个条件的子序列,x,y可以为零。
数据范围如下,比较重要:
基本思路:
我们首先要使用vector<int> memo[201]
数组从前往后记录一下每个数在数组中出现的每个位置,同时记录,在每个位置 i
之前当前数a[i]
的出现次数;
这个时候我们遍历一遍数组,对于每个位置 i
的数 a[i]
,和之前的这个数出现的次数x
,我们在memo
数组里可以O(1)找到从后往前a[i]
出现数目也为x的后半段位置,然后在这两个前后位置之间,找出现次数最多的数作为y
就行了;
在前后两个位置之间找出现次数最多的数,我们可以再在之前预处理一个200个数的前缀和数组,然后同样O(1)查询就是了。
PS.这题的空间卡的很死,所以我们数组不要多开太多,空间卡紧一点。
参考代码:
#pragma GCC optimize(2)
#pragma GCC optimize(3)
#include <bits/stdc++.h>
using namespace std;
#define IO std::ios::sync_with_stdio(false)
#define ll long long
#define rep(i, l, r) for (int i = l; i <= r; i++)
#define per(i, l, r) for (int i = l; i >= r; i--)
#define mset(s, _) memset(s, _, sizeof(s))
#define pb push_back
#define pii pair <int, int>
#define pdd pair <double, double>
#define mp(a, b) make_pair(a, b)
#define INF 1e18
inline int read() {
int x = 0, neg = 1; char op = getchar();
while (!isdigit(op)) { if (op == '-') neg = -1; op = getchar(); }
while (isdigit(op)) { x = 10 * x + op - '0'; op = getchar(); }
return neg * x;
}
inline void print(int x) {
if (x < 0) { putchar('-'); x = -x; }
if (x >= 10) print(x / 10);
putchar(x % 10 + '0');
}
const int maxn = 2e5 + 3;
int memo[maxn][201];
vector<int> vec[201];
int n,a[maxn],c[maxn];
signed main() {
IO;
int t;
cin >> t;
while (t-- > 0) {
cin >> n;
rep(i, 1, n) {
rep(j, 1, 200) {
memo[i][j] = 0;
vec[j].clear();
}
}
rep(i, 1, n) {
cin >> a[i];
vec[a[i]].push_back(i);
c[i] = vec[a[i]].size();
for (int j = 1; j <= 200; j++) {
memo[i][j] = memo[i - 1][j];
}
memo[i][a[i]]++;
}
int ans = 1;
rep(i, 1, n) {
int s = a[i], x = c[i];
if (vec[s].size() < x * 2) continue;
int ns = vec[s].size();
int r = vec[s][ns - x];
int y = 0;
rep(j, 1, 200) {
y = max(y, memo[r - 1][j] - memo[i][j]);
}
ans = max(ans, y + x * 2);
}
cout << ans << '\n';
}
return 0;
}