codeforces818D Multicolored Cars -- 线段树上二分

用线段树记录当前每种颜色的出现次数,枚举每辆车更新。
当颜色是 m 时在线段树上二分,将出现次数比 m 的出现次数小的颜色标记为1。
没有被标记过的颜色就是答案了。

代码:

#include
#include
#include
#include
using namespace std;
#define N 100010
struct Node{
    int w,f;
}l[N];
int i,j,k,n,m,x,a[N],c[N<<2],w[N<<4],Num,y;
bool b[N];
inline bool Cmp(Node a,Node b){
    return a.winline int Min(int x,int y){return xinline void Update(int x,int l,int r,int y){
    if(l==r){
        c[x]++;
        return;
    }
    int Mid=l+r>>1;
    if(y<=Mid)Update(x<<1,l,Mid,y);else Update(x<<1|1,Mid+1,r,y);
    c[x]=Min(c[x<<1],c[x<<1|1]);
}
inline int Find(int x,int l,int r){
    if(l==r){
        c[x]=n+1;
        return l;
    }
    int Mid=l+r>>1,y;
    if(c[x<<1]1|1])y=Find(x<<1,l,Mid);else y=Find(x<<1|1,Mid+1,r);
    c[x]=Min(c[x<<1],c[x<<1|1]);
    return y;
}
inline void Solve(int x){
    while(c[1]1,1,Num)]=1;
}
int main(){
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++)scanf("%d",&l[i].w),l[i].f=i;
    sort(l+1,l+n+1,Cmp);
    a[l[1].f]=Num=1;w[1]=l[1].w;
    for(i=2;i<=n;i++)
    if(l[i].w==l[i-1].w)a[l[i].f]=Num;else a[l[i].f]=++Num,w[Num]=l[i].w;
    for(i=1;i<=Num;i++)if(w[i]==m){m=i;break;}
    if(i>Num){
        if(m>1)puts("1");else puts("2");
        return 0;
    }
    for(i=1;i<=n;i++){
        Update(1,1,Num,a[i]);
        if(a[i]==m)Solve(++k);
    }
    for(i=1;i<=Num;i++)
    if(i!=m&&!b[i]){
        printf("%d\n",w[i]);
        return 0;
    }
    puts("-1");
    return 0;
}

你可能感兴趣的:(线段树)