HDU 4288 Coder(vector + upper_bound, lower_bound)

题目链接：HDU 4288 Coder

【题目大意】

给你一个空set ，有三种操作，

add x set中加入一个元素x （题目保证set中之前没有x）

del x 删除元素 x （题目保证set中一定有x）

sum （set 中元素 为 a1 a2 a3...... an 求所有下标为i ， 且 i%5==3 的 元素的和）

你要确保输入的元素在set中是有序的。

乍一看直接用set做了， 然而数据规模有 1e5，超时。

比赛的时候偶然听到有人说用vector ， 才来了灵感，用lower_bound找到元素的位置再进行插入删除操作。当然网上说直接用数组暴力，线段树做也可以。

【源代码】

#include <iostream>
#include <sstream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <vector>
#include <set>
using namespace std;
vector<int> a;
int main(){
int n, v, pos;
long long ans; //注意要用longlong
char cmd[10];
while(scanf("%d", &n) != EOF){
a.clear();
while(n--){
scanf("%s", cmd);
if(!strcmp(cmd, "add")){
scanf("%d", &v);
if(a.empty())
a.push_back(v);
else{
a.insert(upper_bound(a.begin(),a.end(),v),v); //找到元素应该插入的位置。
}
}
else if(!strcmp(cmd, "del")){
scanf("%d", &v);
a.erase(lower_bound(a.begin(),a.end(),v)); //找到需要删除元素所在的位置
}
else{
ans = 0;
if(a.size() == 0){
printf("0\n");
}
else{
for(unsigned i = 0; 5*i+2 < a.size(); i++){
ans += a[5*i+2];
}
printf("%lld\n",ans);
}

}
}
}
return 0;
}