hdu 4288セグメントツリー+オフライン+離散化
2943 ワード
http://acm.hdu.edu.cn/showproblem.php?pid=4288
最初は、断固たるTLEで、線分の木の上で%5=3の座標だけを維持して、例えば1 2 3 4 5 6 3番目の数を削除すると、3,6の位置を全+1にして、右にシフトしていると思いますが、和を求めるのは非常に遅いので、10秒でもTLEです..
正しいやり方:1、ノード内メンテナンスsum[0...4]はそれぞれ区間内%5=iの和を表す
2、ノードメンテナンスポイントの個数、cnt
3、離散化処理、そして挿入するたびに通過するノードcnt+1または-1、リーフノードSum[0]はノード値であり、親ノードのsumはこのように維持した:非常に良い方法
4、離散化+オフラインで、削除数を便利にすることができ、mapを使わない
新学会の離散化方法:
最初は、断固たるTLEで、線分の木の上で%5=3の座標だけを維持して、例えば1 2 3 4 5 6 3番目の数を削除すると、3,6の位置を全+1にして、右にシフトしていると思いますが、和を求めるのは非常に遅いので、10秒でもTLEです..
正しいやり方:1、ノード内メンテナンスsum[0...4]はそれぞれ区間内%5=iの和を表す
2、ノードメンテナンスポイントの個数、cnt
3、離散化処理、そして挿入するたびに通過するノードcnt+1または-1、リーフノードSum[0]はノード値であり、親ノードのsumはこのように維持した:非常に良い方法
void pushup(int rt)
{
for(int i=0;i<5;i++)
nodes[rt].sum[i]=nodes[ls(rt)].sum[i]+nodes[rs(rt)].sum[ ((i-nodes[ls(rt)].cnt)%5+5)%5 ];
}
4、離散化+オフラインで、削除数を便利にすることができ、mapを使わない
新学会の離散化方法:
for(int i=0;i<n;i++)
{
scanf("%d",&a[i]);
tmp[num++]=a[i];
}
sort(tmp,tmp+num);
num=unique(tmp,tmp+num)-tmp;
//
int pos=lower_bound(tmp,tmp+num,a[i])-tmp;
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <set>
#include <map>
using namespace std;
#define IN(s) freopen(s,"r",stdin)
#define CL(a,b) memset(a,b,sizeof(a))
#define ll long long
#define ls(rt) rt*2
#define rs(rt) rt*2+1
const int MAXN = 1e5+100;
struct Node{
int l,r;
ll sum[5]; //
int cnt; //
}nodes[MAXN*4];
int a[MAXN],n,tmp[MAXN];
char op[MAXN][6];
void build(int rt,int l,int r)
{
nodes[rt].l=l;
nodes[rt].r=r;
nodes[rt].cnt=0;
CL(nodes[rt].sum,0);
if(l==r)return;
int mid=(l+r)/2;
build(ls(rt),l,mid);
build(rs(rt),mid+1,r);
}
void pushup(int rt)
{
for(int i=0;i<5;i++)
nodes[rt].sum[i]=nodes[ls(rt)].sum[i]+nodes[rs(rt)].sum[ ((i-nodes[ls(rt)].cnt)%5+5)%5 ];
}
void update(int rt, int pos, int d, int flag)
{
nodes[rt].cnt+=flag;
if(nodes[rt].l == nodes[rt].r)
{
nodes[rt].sum[0]+=d;
return;
}
if(pos<=nodes[ls(rt)].r)update(ls(rt),pos,d,flag);
else update(rs(rt),pos,d,flag);
pushup(rt);
}
int main()
{
//IN("hdu4288.txt");
int num;
while(~scanf("%d",&n))
{
num=0;
for(int i=0;i<n;i++)
{
scanf("%s",op[i]);
if(op[i][0]!='s')
{
scanf("%d",&a[i]);
tmp[num++]=a[i];
}
}
sort(tmp,tmp+num);
num=unique(tmp,tmp+num)-tmp;
build(1,1,num);
for(int i=0;i<n;i++)
{
int pos=lower_bound(tmp,tmp+num,a[i])-tmp;
if(op[i][0] == 'a')update(1,pos,a[i],1);
if(op[i][0] == 'd')update(1,pos,-a[i],-1);
if(op[i][0] == 's')printf("%I64d
",nodes[1].sum[2]);
}
}
return 0;
}