Problem - 1335E2 - Codeforces(暴力+优化)

Problem - 1335E2 - Codeforces

题目大意： 给定你长度为$n$的序列，让你将其分成三个子序列，满足如下关系$[\underset{\text{x}}{\underbrace{a,...,a}},\underset{\text{y}}{\underbrace{b,...,b}},\underset{\text{x}}{\underbrace{a,...,a}}]$。

解题思路：对于$a_i$的值范围比较小，所以可以考虑用前缀和来维护区间上对应数字的个数，然后不断去枚举，因为对于给定的$a_i$的值，左边一部分的$a_i$个数一定需要等于右边一部分的$a_i$个数，双指针不断向里面推进就可以，但是因为数据量比较大，如果单纯的暴力的话时间复杂度是$O(200\ast 200\ast n)$,所以对于每个$a_i$，我们只存其对应的下标，这样可以确保每次双指针向内推进的时候都是有效值，这样摊还下来复杂度就降到了$O(200\ast n)$.

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define syncfalse ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
const int N = 2e5+5;
int a[N],pre[N][205];
vector<int>id[205];
void solve(){
    int n;
    cin>>n;
    a[n+1]=0;
    for (int i = 1; i <= 200; ++i)id[i].clear();
    for (int i = 0; i <= n+1; ++i){
        for (int j = 0; j <= 200; ++j){
            pre[i][j]=0;
        }
    }
    for (int i = 1; i <= n; ++i){
        cin>>a[i];
        id[a[i]].push_back(i);
    }
    for (int i = 1; i <= n+1; ++i){
        for (int j = 1; j <= 200; ++j){
            pre[i][j]+=pre[i-1][j];
            if (a[i]==j)pre[i][j]++;
        }
    }
    int ans = 0;
    for (int i = 1; i <= 200; ++i){
        ans=max(ans, pre[n+1][i]-pre[0][i]);
    }
    for (int j = 1; j <= 200; ++j){
        int lid = 0, rid = id[j].size()-1,val=1;
        while(lid<rid){
            int l = id[j][lid], r=id[j][rid];
            for (int i = 1; i <= 200; ++i){
                int tem = pre[r-1][i]-pre[l][i];
                ans=max(ans, tem+2*val);
            }
            lid++,rid--;val++;
        }
    }
    cout << ans << "\n";
}

int main(){
    syncfalse
    #ifndef ONLINE_JUDGE
    freopen("in.txt","r",stdin);
    #endif
    int t;
    cin>>t;
    for (;t;--t){
        solve();
    }

    return 0;
}