hdu 4288 Coder（线段树＋离散化，lower_bound&&upper_bound的运用）

题目链接：

http://acm.hdu.edu.cn/showproblem.php?pid=4288

解题思路：

题目大意：

ｎ次操作（ｎ<=10^5），三种类型：
1.add x 向集合中添加元素x。
2.del x 从集合中删除元素x。
3.sum 询问集合中元素从小到大排列时的下标i % 5 == 3的数之和。
题目保证每个元素唯一。且元素大小不超过10^9。

算法思想：

不知道该怎么说了，按题目意思来就行了，不过，用线段树做时，要事先离散化。。。

AC代码（线段树＋离散化）：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

typedef long long ll;
const int maxn = 100010;

struct node{
    ll mod[5];
}tree[maxn<<2];
int num[maxn<<2],tab[maxn],a[maxn];
char op[maxn][10];

void build(int id,int l,int r){
    num[id] = 0;
    for(int i = 0; i < 5; i++)
        tree[id].mod[i] = 0;
    if(l == r)
        return;
    int mid = (l+r)>>1;
    build(id<<1,l,mid);
    build(id<<1|1,mid+1,r);
}

void pushup(int id){
    num[id] = num[id<<1]+num[id<<1|1];
    for(int i = 0; i < 5; i++)
        tree[id].mod[i] = tree[id<<1].mod[i];
    int lx = num[id<<1];
    for(int i = 0; i < 5; i++)
        tree[id].mod[(i+lx)%5] += tree[id<<1|1].mod[i];
}

void update(int id,int l,int r,int p,int val){
    if(l == r){
        num[id] = val;
        tree[id].mod[1] = val*tab[l];
        return;
    }
    int mid = (l+r)>>1;
    if(p <= mid)
        update(id<<1,l,mid,p,val);
    else
        update(id<<1|1,mid+1,r,p,val);
    pushup(id);
}

int bin(int l,int r,int key){
    while(l <= r){
        int mid = (l+r)>>1;
        if(tab[mid] == key)
            return mid;
        else if(tab[mid] > key)
            r = mid-1;
        else
            l = mid+1;
    }
}

int main(){
    int n,cnt;
    while(~scanf("%d",&n)){
        cnt = 0;
        for(int i = 0; i < n; i++){
            scanf("%s",op[i]);
            if(op[i][0] != 's'){
                scanf("%d",&a[i]);
                if(op[i][0] == 'a')
                    tab[++cnt] = a[i];
            }
        }
        sort(tab+1,tab+cnt+1);
        int k = 1;
        for(int i = 2; i <= cnt; i++)
            if(tab[i] != tab[k])
                tab[++k] = tab[i];
        build(1,1,k);
        for(int i = 0; i < n; i++){
            if(op[i][0] == 'a'){
                int pos = bin(1,k,a[i]);
                update(1,1,k,pos,1);
            }
            else if(op[i][0] == 'd'){
                int pos = bin(1,k,a[i]);
                update(1,1,k,pos,0);
            }
            else
                printf("%lld\n",tree[1].mod[3]);
        }
    }
    return 0;
}

AC代码（lower_bound&&upper_bound的运用）：

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;

typedef long long ll;
vector<int> v;
vector<int>::iterator it;

int main(){
    int n,x;
    while(~scanf("%d",&n)){
        v.clear();
        char op[10];
        while(n--){
            scanf("%s",op);
            if(op[0] == 'a'){
                scanf("%d",&x);
                it = upper_bound(v.begin(),v.end(),x);
                v.insert(it,x);
            }
            else if(op[0] == 'd'){
                scanf("%d",&x);
                it = lower_bound(v.begin(),v.end(),x);
                v.erase(it);
            }
            else{
                ll sum = 0;
                for(int i = 2; i < v.size(); i += 5)
                    sum += v[i];
                printf("%lld\n",sum);
            }
        }
    }
    return 0;
}