3658 小明统计数组
小明最近喜欢上数组统计,但是他遇到了一道难题,题目如下:
最初数列是空的,那么一共有下面五种操作:
1. a x 表示向数列里面增加数值为x的数;
2. d x 表示删除数列里面数值为x的数;
3. I 表示删除数列里面数值最大的数;
4. m 表示删除数列里面数值最小的数;
5. q 询问当前数列的和。
保证添加的元素都不不一样,对于删除操作,如果数列里面不存在这个元素,则不进行删除操作。小明不会做这个题,希望寻求你的帮助,聪明的你可以帮助小明解决这个问题吗?
在处理输入的时候要特别注意!题目中操作 3 的操作字符是 大写的 i,而不是小写的 l。
输入
第一行输入两个数n,表示操作的数目;
之后n行,每行为题目中描述的五种操作。
输出
对于每个询问,输出当前数列的和。
数据范围
1<=n<=100000
1<=x<=100000且x互不相同
输入样例
11
a 1
d 2
q
a 2
a 3
a 4
q
I
q
m
q
输出样例
1
10
6
5
样例解释
a 1 增加1到数列里面
d 2 删除数列里面数值为2的元素,因为数列里面没有,所以不进行操作
q 询问,当前数列的和为1
a 2 增加2到数列里面
a 3 增加3到数列里面
a 4 增加4到数列里面
q 询问,当前数列的和为10
I 删除数列里面数值最大的元素,删除4
q 询问,当前数列的和为6
m 删除数列里面数值最小的元素,删除1
q 询问,当前数列的和为5
解析:
由于题目中每个元素都是不同的,且有删除最大值和最小值的操作,所以我们可以用一个 set 来维护这个数列,时间复杂度为O(nlogn)。
set 可以直接删除具体的元素,因为 set是有序的,所以最小值是第一个,最大值是最后一个,直接进行删除即可,我们可以用一个变量来维护当前数列元素的总和。
放代码:
#include<bits/stdc++.h>
using namespace std;
int main(){
set<int> s;
int t, sum = 0; cin >> t;
while(t--){
char op; cin >> op;
int x;
if(op == 'a'){
cin >> x;
s.insert(x);
sum += x;
}
else if(op == 'd'){
cin >> x;
if(s.find(x) != s.end()){
s.erase(x);
sum -= x;
}
}
else if(op == 'm'){
if(!s.empty()){
sum -= *(s.begin());
s.erase(*s.begin());
}
}
else if(op == 'I'){
if(!s.empty()){
sum -= *(--s.end());
s.erase(*--s.end());
}
}
else if(op == 'q'){
printf("%d\n", sum);
}
}
return 0;
}