D. Multiset(思维+树状数组二分/权值线段树)

https://codeforces.com/problemset/problem/1354/D


思路:找排名实际上就是看每个数前面的数出现的次数累加。如果用前缀和不得行。

于是转化到值上去。开一个值域树状数组。对于一个排名,我二分里面的数,logn获得<=这个数的出现次数总和。单调性可以调整。

O(n*logn*logn)

当然权值线段树也行。O(nlogn)

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define lowbit(x) x&(-x)
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=1e6+10;
typedef int LL;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
LL tree[maxn];
LL n,q;
LL add(LL x,LL d){
    while(x<=n){
        tree[x]+=d;
        x+=lowbit(x);
    }
}
LL query(LL x){
    LL sum=0;
    while(x>0){
        sum+=tree[x];
        x-=lowbit(x);
    }return sum;
}
int main(void){
   n=read();q=read();
   for(LL i=1;i<=n;i++){
      LL x;x=read();
      add(x,1);
   }
   while(q--){
      LL num;num=read();
      if(num>0) add(num,1);
      else{
        num=-num;///排名---出现的第几个
        LL l=1;LL r=n+1;
        while(l<r){
            LL mid=(l+r)>>1;///二分排名对应的数字
            if(query(mid)>=num) r=mid;
            else l=mid+1;
        }
        if(l==n+1) continue;
        add(l,-1);
      }
   }
   for(LL i=1;i<=n;i++){
      if(query(i)!=0){
        printf("%d\n",i);
        return 0;
      }
   }
   printf("0\n");
   return 0;
}

猜你喜欢

转载自blog.csdn.net/zstuyyyyccccbbbb/article/details/115356875