[ABC212D] Querying Multiset 题解

[ABC212D] Querying Multiset

Description

给你一个集合，让你支持三种操作：

• 将 \(x\) 加入集合。

• 把集合中的数都加上 \(x\)。

• 将集合中最小的数删除，并且输出这个数。

---

Solution

考虑使用优先队列维护这个集合。

注意到操作 2 这个区间加比较难维护，可以考虑设置一个增加量 \(s\) 来代表所有操作 2 中增加的数。

那么在操作 1 中，输入的 \(x\) 没有增加过 \(s\) ，所以直接 push(x-s)。

在操作 2 中比较简单，直接把 \(x\) 加到 \(s\) 里即可。

操作 3 ，直接输出 \(x+s\)。

然后就做完了，复杂度 \(O(T\log |S|)\)。

#include<bits/stdc++.h>
#define int long long
using namespace std;
long long T,opt,x;
long long diff=0;
priority_queue<int>pq;
signed main(){ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);cin>>T;while(T--){cin>>opt;if(opt==1){cin>>x;pq.push(diff-x);}else if(opt==2){cin>>x;diff+=x;}else{cout<<diff-pq.top()<<endl;pq.pop();}}return 0;
}