自己一开始yy的只要是建新树的时候用lazytag,然后tag不往子节点传,而是到这里就新生成2个子节点。
然而这样会TLE,我也不知道为什么。然后我加了个小优化,从lastnode[tm]表示到tm为止的节点数,然后回到tm时,tot=lastnode[tm]+1;
然而并没卵用,TLE
主席树的数组开太大又会MLE。。。。
后来上网学习了一波神奇姿势。tag就记着放那里不要传递,在求和的时候顺路就把tag*(r-l+1)算在里面。发现网上没让tot回来,是不是这个小优化错了?不知道,先去掉,TLE。
好的,按照神奇姿势改好,TLE,卧槽你T个寂寞啊,一直T,我这检查几十遍了还T。
不管了,把那个小优化加回来,交一发。。喝口水,恶膜某民命秒没。。。卧槽A了?差点喷一电脑的水,还好喝下去了。原来TLE是数组越界了啊。网上那个人比我开的大了一点点就A了,都没有小优化。。卧槽,HDU的评测回复没有CE这种东西吗?很烦
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define maxl 100010
using namespace std;
struct node
{
int lnode,rnode,tag;
long long sum;
}tree[maxl*20];
int n,m,tim,tot;
int a[maxl],rt[maxl],lastnode[maxl];
char ch[2];
void build(int k,int l,int r)
{
tree[k].sum=0;tree[k].tag=0;
if(k>tot) tot=k;
if(l==r)
{
tree[k].sum=a[l];
tree[k].lnode=0;tree[k].rnode=0;
return;
}
int mid=(l+r)>>1;
tree[k].lnode=k<<1;tree[k].rnode=k<<1|1;
build(k<<1,l,mid);
build(k<<1|1,mid+1,r);
tree[k].sum=tree[k<<1].sum+tree[k<<1|1].sum;
}
void prework()
{
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
tot=1;
build(1,1,n);
tim=1;lastnode[1]=tot;rt[1]=1;
tot++;
}
/*void gank(int k)
{
if(tree[k].l==tree[k].r)
return;
int l=tree[k].lnode,r=tree[k].rnode;long long x=tree[k].tag;
if(tree[k].tag)
{
tree[k].lnode=tot++;
tree[tot-1]=tree[l];
tree[tot-1].sum+=(tree[l].r-tree[l].l+1)*x;
tree[tot-1].tag=x;
tree[k].rnode=tot++;
tree[tot-1]=tree[r];
tree[tot-1].sum+=(tree[r].r-tree[r].l+1)*x;
tree[tot-1].tag=x;
tree[k].tag=0;
}
}*/
long long query(int k,int l,int r,int l1,int r1)
{
int mid=(l1+r1)>>1;
long long ans=(r-l+1)*(long long)tree[k].tag;
if(l1==l && r1==r)
return tree[k].sum;
if(l>mid)
return ans+query(tree[k].rnode,l,r,mid+1,r1);
else if(r<=mid)
return ans+query(tree[k].lnode,l,r,l1,mid);
else
return ans+query(tree[k].lnode,l,mid,l1,mid)+query(tree[k].rnode,mid+1,r,mid+1,r1);
}
void change(int d,int &x,int l,int r,int l1,int r1)
{
tree[tot++]=tree[x];x=tot-1;
lastnode[tim]=x;
tree[x].sum+=(r-l+1)*(long long)d;
if(l==l1 && r==r1)
{
tree[x].tag+=d;
return;
}
int mid=(l1+r1)>>1;
if(l>mid)
change(d,tree[x].rnode,l,r,mid+1,r1);
else if(r<=mid)
change(d,tree[x].lnode,l,r,l1,mid);
else
change(d,tree[x].lnode,l,mid,l1,mid),
change(d,tree[x].rnode,mid+1,r,mid+1,r1);
}
void mainwork()
{
int l,r,t,d;
for(int i=1;i<=m;i++)
{
scanf("%s",ch);
if(ch[0]=='Q')
{
scanf("%d%d",&l,&r);
printf("%lld\n",query(rt[tim],l,r,1,n));
}
else if(ch[0]=='H')
{
scanf("%d%d%d",&l,&r,&t);
printf("%lld\n",query(rt[t+1],l,r,1,n));
}
else if(ch[0]=='C')
{
scanf("%d%d%d",&l,&r,&d);
tim++;rt[tim]=rt[tim-1];
change(d,rt[tim],l,r,1,n);
}
else
{
scanf("%d",&t);
tot=lastnode[t+1]+1;
tim=t+1;
}
}
}
void print()
{
// printf("\n");
}
int main()
{
tree[0].lnode=tree[0].rnode=tree[0].sum=0;
rt[0]=0;lastnode[0]=0;
while(~scanf("%d%d",&n,&m))
{
prework();
mainwork();
print();
}
return 0;
}