Educational Codeforces Round 87 (Rated for Div. 2) D. Multiset 题解（思维+二分）

题目链接

题目思路

我用的是vector加二分，但是t了，后面了解到rease和insert的操作是o（n）的复杂度，感觉以后这个操作没什么用了。。

看了一个神仙二分，判断每个数相对于这个多重集合的大小，二分这个集合中最小的元素。

还有我居然以为1e6在28mb放不下我少算了一个0。。。

tle代码

#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
int n,q;
priority_queue<int,vector<int>,greater<int> >pq;
vector<int> vec;
int main(){
    scanf("%d %d",&n,&q);
    for(int i=1,x;i<=n;i++){
        scanf("%d",&x);
        pq.push(x);
    }
    while(!pq.empty()){
        vec.push_back(pq.top());
        pq.pop();
    }
    for(int i=1,opt;i<=q;i++){
        scanf("%d",&opt);
        if(opt<0){
            vec.erase(vec.begin()-opt-1);
        }else{
            vec.insert(lower_bound(vec.begin(),vec.end(),opt),opt);
        }
    }
    if(!vec.empty()){
        printf("%d\n",vec[0]);
    }else{
        printf("0\n");
    }
    return 0;
}

二分代码

#include<cstdio>
using namespace std;
const int maxn=1e6+5;
typedef long long ll;
int n,q,a[maxn],k[maxn];
bool check(int x){
    int cnt=0;//这个数在这个集合里面排多少位
    for(int i=1;i<=n;i++){
        cnt+=(x>=a[i]);
    }
    for(int i=1;i<=q;i++){
        if(k[i]<0){//删除-k[i]大的数
            cnt-=(cnt>=-k[i]);
        }else{//加入k[i]
            cnt+=(x>=k[i]);
        }
    }
    return cnt>0;
}
int main(){
	scanf("%d %d",&n,&q);
	for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
	}
	for(int i=1;i<=q;i++){
        scanf("%d",&k[i]);
	}
	int l=1,r=1e6,ans=0;
	while(l<=r){
        int mid=(l+r)/2;
        if(check(mid)){
            ans=mid;
            r=mid-1;
        }else{
            l=mid+1;
        }
	}
	printf("%d\n",ans);
	return 0;
}