HDU 4288 && Codeforces 85D (线段树+离散化+离线处理)

来源：互联网发布：网络直播培训编辑：程序博客网时间：2024/06/11 03:49

首先，为什么要离散化和离线处理？因为输入的数字会很大，可以达到10^9，但输入的N最多为10^5，即做多添加100000个数(不删除任何数)，所以把输入的值进行排序，离散成1~100000，同时这也可以知道要建多大的线段树。

离散化的时候，经常要用到去重操作和查找操作，一般查找是用二分查找的。通过这题学习到了两个新的STL。

1)unique:去重 2)lower_bound:返回第一个比需要查找的数大的或者相等的数组下标（相当于二分查找）。具体实现在代码中。

最后，这题的pushUp操作是精髓，自己就是想不出来，唉。

#include <iostream>#include <cstdio>#include <cstring>#include <algorithm>#define SIZE 100005#define ls l,mid,rt<<1#define rs mid+1,r,rt<<1|1using namespace std;typedef __int64 Int;Int sum[SIZE<<2],ans[SIZE<<2][5];Int arr[SIZE],der[SIZE];char op[SIZE][5];int n;void pushUp(int rt){    for(int i=0; i<5; i++)        ans[rt][i] = ans[rt<<1][i] + ans[rt<<1|1][(i-sum[rt<<1]%5+5)%5];}void build(int l,int r,int rt){    sum[rt] = 0;    for(int i=0; i<5; i++)ans[rt][i] = 0;    if(l == r)return;    int mid = (l + r) >> 1;    build(ls);    build(rs);}void update(int l,int r,int rt,int pos,int add){    sum[rt] += add;    if(l == r)    {        ans[rt][1] = add==1?der[pos]:0;        return;    }    int mid = (l + r) >> 1;    if(pos <= mid) update(ls,pos,add);    else update(rs,pos,add);    pushUp(rt);}int main(){    while(~scanf("%d",&n))    {        int len = 0;        for(int i=0; i<n; i++)        {            scanf("%s",op[i]);            if(op[i][0] != 's')            {                scanf("%I64d",&arr[i]);                der[len++] = arr[i];            }        }        if(n == 1)        {            if(op[0][0] == 's')                puts("0");            continue;        }        sort(der,der+len);        len = unique(der,der+len)-der;        build(1,len,1);        int pos;        for(int i=0; i<n; i++)        {            pos = lower_bound(der,der+len,arr[i])-der;            if(op[i][0] == 's')                printf("%I64d\n",ans[1][3]);            else if(op[i][0] == 'a')                update(1,len,1,pos,1);            else                update(1,len,1,pos,-1);        }    }    return 0;}