E2. Three Blocks Palindrome (hard version)(枚舉+思維+前綴和)

在這裏插入圖片描述
在這裏插入圖片描述
題目大意:
給你一個長度爲n的序列,要你找出形如aba這樣的迴文,其中a,b 可以等於0,並且最多隻能出現兩個不同字符。
思路:
注意到字符很少,只有200,可以從這裏下手,先把[1,200]每個字符的前綴和求出來,再把相同字符的位置存起來,然後枚舉200個數,再枚舉每個數的長度的一半,然後在當前枚舉數的中間找一個最多的一個單的字符這樣就行了。需要預處理一下b=0這種情況,也就在只有一個字符時最多的是哪個字符。
代碼:

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

const int maxn = 2e5 + 10;
int sum[maxn][201];
void solved() {
	int t;
	for (scanf("%d", &t); t; t--) {
		int n; cin >> n;
		vector<int>a; a.push_back(0);
		vector<int>ve[206];
		for (int i = 1; i <= n; i++)
			for (int j = 1; j <= 200; j++)
				sum[i][j] = 0;
		for (int i = 1; i <= n; i++) {
			int x; cin >> x; a.push_back(x);
			ve[x].push_back(i);
		}
		for (int i = 1; i <= n; i++) {
			for (int j = 1; j <= 200; j++) {
				if (a[i] == j)
					sum[i][j] = sum[i - 1][j] + 1;
				else sum[i][j] = sum[i - 1][j];
			}
		}
		int ans = 0;
		for (int j = 1; j <= 200; j++)
			ans = max(ans, sum[n][j]);
		for (int i = 1; i <= 200; i++) {
			for (int j = 1; j <= ve[i].size() / 2; j++) {
				int M = 0;
				int l = ve[i][j - 1], r = ve[i][ve[i].size() - j];
				for (int k = 1; k <= 200; k++) {
					M = max(M, sum[r - 1][k] - sum[l][k]);
				}
				ans = max(ans, M + 2 * j);
			}
		}
		cout << ans << endl;
	}
}
int main() {
	solved();
	return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章