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 = 100010;
typedef long long ll;
int a[maxn];

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

也可以用线段树来维护。先把所有的数据都读进来，然后离散化一下建树。树的每个节点维护一个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 = 100010;
typedef long long ll;
struct Tree {
int cnt;
ll sum[5];
}tree[maxn << 2];
int dat[maxn], tmp[maxn];
char cmd[maxn][10];
void pushup(int rt)
{
for (int i = 0; i < 5; i++)
tree[rt].sum[i] = tree[rt<<1].sum[i] + tree[rt<<1|1].sum[((i - tree[rt<<1].cnt) % 5 + 5) % 5];
tree[rt].cnt = tree[rt<<1].cnt + tree[rt<<1|1].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 = 0; i < 5; i++) tree[rt].sum[i] = 0;
tree[rt].cnt = 0;
if (l == r) return;
int mid = (l + r) / 2;
build(rt<<1, l, mid);
build(rt<<1|1, mid + 1, 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[0] = val;
return;
}
int mid = (l + r) / 2;
if (p <= mid) update(rt<<1, l, mid, p, val, flag);
else update(rt<<1|1, mid + 1, r, p, val, flag);
pushup(rt);
}
int main()
{
int n;
while (~scanf("%d", &n))
{
int num = 0;
for (int i = 0; i < n; i++) {
scanf("%s", cmd[i]);
if (cmd[i][0] != 's')
{
scanf("%d", &dat[i]);
tmp[num++] = dat[i];
}
}
sort(tmp, tmp + num);
num = unique(tmp, tmp + num) - tmp;

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