HDU 4288 Coder(线段树)

题意：

给定三种操作

　　1. add x 向序列中添加x，添加之后序列还保持有序

　　2. del x  删除序列中值为x的元素

　　3. sum  求下边模5等于3的元素和

思路：

直接暴力也可以过，就是看暴力写的好不好了。用数组直接暴力可过。

暴力代码：

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
const int maxn = ;
typedef long long ll;
int a[maxn];

int main()
{
    int n;
    while (~scanf("%d", &n))
    {
        char cmd[];
        int d;
        int len = ;
        while (n--)
        {
            scanf("%s", cmd);
            if (cmd[] == 'a')
            {
                scanf("%d", &d);
                int i;
                for (i = len++; i >= ; i--)
                {
                    if (a[i] <= d) break;
                    a[i + ] = a[i];
                }
                a[i + ] = d;
            }
            else if (cmd[] == 'd')
            {
                scanf("%d", &d);
                int i;
                for (i = ; i <= len; i++)
                    if (a[i] == d) break;
                for (; i < len; i++)
                    a[i] = a[i + ];
                len--;
            }
            else
            {
                ll ans = ;
                for (int i = ; i <= len; i += )
                    ans += (ll)a[i];
                printf("%lld\n", ans);
            }
        }
    }
    return ;
}

也可以用线段树来维护。先把所有的数据都读进来，然后离散化一下建树。树的每个节点维护一个sum[5]，就是模上5之后的余数，还有一个当前区间的有多少个数字的个数。注意：每个节点的sum当中的下边都是对应线段树中该节点的下边来说的。所以叶子节点一定对应1，就是sum[0], 关键就是pushup，因为往上更新的时候，考虑一个父区间，它的左孩子区间对应的sum[i]，就是父区间的sum[i]，但是右区间的sum[i]对应的不是父区间的sum[i],而是sum[(i + lson.cnt)%5]。所以这样就可以做拉。

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
const int maxn = ;
typedef long long ll;
struct Tree {
    int cnt;
    ll sum[];
}tree[maxn << ];
int dat[maxn], tmp[maxn];
char cmd[maxn][];
void pushup(int rt)
{
    for (int i = ; i < ; i++)
        tree[rt].sum[i] = tree[rt<<].sum[i] + tree[rt<<|].sum[((i - tree[rt<<].cnt) %  + ) % ];
    tree[rt].cnt = tree[rt<<].cnt + tree[rt<<|].cnt;
    //当然也可以这么写。
    //int j = (i + tree[rt<<1].cnt) % 5;
    //tree[rt].sum[j] = tree[rt<<1].sum[j] + tree[rt<<1|1].sum[i];
}
void build(int rt, int l, int r)
{
    for (int i = ; i < ; i++) tree[rt].sum[i] = ;
    tree[rt].cnt = ;
    if (l == r) return;
    int mid = (l + r) / ;
    build(rt<<, l, mid);
    build(rt<<|, mid + , r);
}
void update(int rt, int l, int r, int p, int val, int flag)
{
    if (l == r)
    {
        tree[rt].cnt += flag;
        tree[rt].sum[] = val;
        return;
    }
    int mid = (l + r) / ;
    if (p <= mid) update(rt<<, l, mid, p, val, flag);
    else update(rt<<|, mid + , r, p, val, flag);
    pushup(rt);
}
int main()
{
    int n;
    while (~scanf("%d", &n))
    {
        int num = ;
        for (int i = ; i < n; i++) {
            scanf("%s", cmd[i]);
            if (cmd[i][] != 's')
            {
                scanf("%d", &dat[i]);
                tmp[num++] = dat[i];
            }
        }
        sort(tmp, tmp + num);
        num = unique(tmp, tmp + num) - tmp;

        build(, , n);
        for (int i = ; i < n; i++)
        {
            int p = lower_bound(tmp, tmp + num, dat[i]) - tmp;
            if (cmd[i][] == 's')
                printf("%lld\n", tree[].sum[]);
            else if (cmd[i][] == 'a')
                update(, , n, p, dat[i], );
            else update(, , n, p, , -);
        }
    }
    return ;
}