天天看点

Codeforces 818D Multicolored Car(思维+线段树)

题意:

现在有N辆车经过,A和B玩一个游戏,他们统计他们自己喜欢的颜色的车辆的个数。

一开始对手已经选好了颜色A.我们现在要选择一个颜色,使得Cnti(chose)>=Cnti(A);

这里Cnti()表示到位子i,该颜色出现的车辆个数。

如果存在多解,输出任意一个可行解即可。

思路:

考虑当前数的贡献:

如果当前数字是A,那么对其他所有数来说,都需要有一个数才能满足,否则这个数直接pass。

因此可以出现A,让其他所有数都-1,如果此时是负数了,那就彻底gg。

然后出现不是A的其他数字,其贡献就是给本身这个次数+1。不过要先判他是不是已经gg了。不gg还能加1,gg了就让他保持负数。

因此构造出一颗线段树,记录每一种颜色都出现了多少次。

遍历这N辆车,当前是A,我们对应在这棵树中所有位子都减1.表示需要其他位子需要有一个数来抵消这个cnt(A);

如果当前!=A,如果此时这个位子不是负数(即之前所有位子都没有出现过A颜色的车辆多于这种颜色的车辆的情况),那么对应这种颜色就可能是一个可行解,那么对应modify(i,i,1);

最终在1e6种颜色,也就是1e6个位置上,找到一个>=0的任意一个颜色就是答案。

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=1e6+100;
typedef long long LL;
struct Tree{
    LL l,r,maxval,minval,sum,tag;///区间最大值,区间最小值,区间和,l
}tree[maxn*4];
void push_up(LL p)
{
    tree[p].maxval=max(tree[p*2].maxval,tree[p*2+1].maxval);
    tree[p].minval=min(tree[p*2].minval,tree[p*2+1].minval);
    tree[p].sum=tree[p*2].sum+tree[p*2+1].sum;
}
void addtag(LL p,LL d)
{
    tree[p].tag+=d;
    tree[p].sum+=d*(tree[p].r-tree[p].l+1);
    tree[p].maxval=tree[p].tag;
    tree[p].minval=tree[p].tag;
}
void push_down(LL p)
{
    if(tree[p].tag!=0){
        addtag(p*2,tree[p].tag);
        addtag(p*2+1,tree[p].tag);
        tree[p].tag=0;
    }
}
void build(LL p,LL l,LL r)
{
    tree[p].l=l;tree[p].r=r;tree[p].maxval=-1e18;tree[p].minval=1e18;
    tree[p].sum=0;tree[p].tag=0;
    if(l==r) {tree[p].maxval=tree[p].minval=tree[p].sum=0;return;}
    LL mid=(l+r)>>1;
    build(p*2,l,mid);
    build(p*2+1,mid+1,r);
    push_up(p);
}
void modify(LL p,LL l,LL r,LL d)
{
    if(l<=tree[p].l&&r>=tree[p].r)///要修改的区间涵盖在区间内部
    {
            addtag(p,d);
            return;
    }
    push_down(p);
    LL mid=(tree[p].l+tree[p].r)>>1;
    if(l<=mid) modify(p*2,l,r,d);
    if(r>mid) modify(p*2+1,l,r,d);
    push_up(p);
}
LL query(LL p,LL l,LL r)
{
    if(l<=tree[p].l&&r>=tree[p].r)
    {
            return tree[p].sum;
    }
    LL ans=0;
    push_down(p);
    LL mid=(tree[p].l+tree[p].r)>>1;
    if(l<=mid) ans+=query(p*2,l,r);
    if(r>mid) ans+=query(p*2+1,l,r);
    return ans;
}
int main(void)
{
  cin.tie(0);std::ios::sync_with_stdio(false);
  LL n,b;cin>>n>>b;
  build(1,1,1000000);
  for(LL i=1;i<=n;i++){
    LL x;cin>>x;
    if(x==b){
        modify(1,1,1000000,-1);
    }
    else{
        if(query(1,x,x)>=0){
            modify(1,x,x,1);
        }
    }
  }
  LL ans=-1;
  for(LL i=1;i<=1000000;i++){
     if(i==b) continue;
     if(query(1,i,i)>=0){
        ans=i;
        break;
     }
  }
  cout<<ans<<"\n";
return 0;
}