bzoj4893项链分赃

结论:答案小于等于颜色数
暴力判断1,前缀和
2,用一个队列维护,使所有颜色数等于一半
拓扑学证明B站有视频

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<queue>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=100010;
int n,smm[N][3],q[3],a[N],c[5];
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]); smm[i][0]=smm[i-1][0];smm[i][1]=smm[i-1][1];smm[i][2]=smm[i-1][2];
        ++smm[i][a[i]];
    }
    for(int i=1;i<=n;i++){
        c[a[i]]++;
        if(c[0]<<1==smm[n][0]&&c[1]<<1==smm[n][1]&&c[2]<<1==smm[n][2]){cout<<"1"<<endl;return 0;}
    }
    int cur=1;
    for(int i=1;i<=n;i++){
        ++q[a[i]];
        for(int j=0;j<=2;j++){
            while(cur<i&&(q[j]<<1)>smm[n][j])q[a[cur]]--,cur++;
        }
        if( ((q[0]<<1)==smm[n][0])&&
            ((q[1]<<1)==smm[n][1])&&
            ((q[2]<<1)==smm[n][2])){
                printf("2\n");
                return 0;
            }
    }
    printf("3\n");
    return 0;
}