ZOJ 3522 Hide and seek

题目大意：大小姐与二小姐玩捉迷藏。红魔馆里有N（1<=N<=50000）个可以

来躲的地方，N-1条双向边连接所有的地方（就是一棵树），芙兰有100000

组询问问如果X,Y之间添加一条长度为L的边，树上所有点到X最短距离变化值

与到Y变化值之和。

看到这题题面跟东方有关就做了……咱用LCT做的。维护一大堆迷之变量： 虚儿

子大小的总和num，所在子树大小siz，所在splay子树上边权和sum，以及

（难以描述）ran和lan。

对于求所有点到t点的距离，先把t点提到根，再把新加的边连到的s点access一

下，在原本的路径上找到一个点p，使得t从原路径到p的距离与t从新路径到p

距离基本相等。认为p点以下的点都受到了新加边的影响，所以每个点认为对

答案造成了（s到t原路径长度-新路径长度）的影响。由于每个点的影响由于位

置不同，会有一些差值。lan和ran就是记录这个差值的。

在把t提到根节点的过程中，边权不太好维护。看到一种方法，就是把边视为

权值为L但不计入siz和num的点，而正常点的权值为0。感到学习到了非常棒

的姿势……

写这个玩意错了好多次，问题主要在找p点和long long……果然咱还是太弱

了。

细节还是有点多，得注意下。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=100100;
int n;
int wea;
int stk[maxn],top;
int beg[maxn],w[maxn<<2][3],e;
void putin(int s,int t,int v)
{
w[++e][0]=t;
w[e][1]=beg[s];
beg[s]=e;
w[e][2]=v;
}
struct LCT{
long long l[maxn],r[maxn],f[maxn],rev[maxn],siz[maxn],sum[maxn],num[maxn],lan[maxn],ran[maxn],val[maxn];
void csh()
{
memset(l,0,sizeof(l));
memset(r,0,sizeof(r));
memset(f,0,sizeof(f));
memset(rev,0,sizeof(rev));
memset(lan,0,sizeof(lan));
memset(ran,0,sizeof(ran));
memset(siz,0,sizeof(siz));
memset(num,0,sizeof(num));
memset(beg,0,sizeof(beg));
memset(val,0,sizeof(val));
e=0;
}
bool isroot(int h)
{
if(!f[h])
return 1;
if(l[f[h]]==h||r[f[h]]==h)
return 0;
return 1;
}
void push_down(int h)
{
if(!rev[h]) return;
long long tmp;
tmp=l[h];l[h]=r[h];r[h]=tmp;
rev[l[h]]^=1;rev[r[h]]^=1;
rev[h]=0;
swap(lan[h],ran[h]);   //开始tmp是int,害咱错了好⑨
}
void push_up(int h)
{
if(h==0) return;
push_down(h);
push_down(l[h]);
push_down(r[h]);//这里的push_down（似乎）很重要
siz[h]=siz[l[h]]+siz[r[h]]+num[h];
sum[h]=sum[l[h]]+sum[r[h]]+val[h];
lan[h]=lan[r[h]] + (sum[l[h]]+val[h])*siz[r[h]] + sum[l[h]]*num[h] + lan[l[h]];
ran[h]=ran[l[h]] + (sum[r[h]]+val[h])*siz[l[h]] + sum[r[h]]*num[h] + ran[r[h]];
}//看push_up基本就可以理解lan和ran的含义了
void rotate(int h)
{
int fa=f[h],gfa=f[fa];
if(fa==l[gfa]) l[gfa]=h;
else if(fa==r[gfa]) r[gfa]=h;
f[h]=gfa;
if(h==l[fa])
{
l[fa]=r[h];f[r[h]]=fa;
r[h]=fa;f[fa]=h;
}
else
{
r[fa]=l[h];f[l[h]]=fa;
l[h]=fa;f[fa]=h;
}
push_up(fa);push_up(h);push_up(gfa);
}
void splay(int h)
{
int fa,gfa;
top=1;
stk[top]=h;
for(int pos=h;!isroot(pos);pos=f[pos])
stk[++top]=f[pos];
while(top)
{
push_down(stk[top]);
top--;
}//printf("%d %d\n",sum[h],f[h]);

while(!isroot(h))
{
fa=f[h];//if(wea==2){printf("?\n");system("pause");}
if(isroot(fa))
{
rotate(h);
return;
}
gfa=f[fa];
if((h==l[fa])^(fa==l[gfa]))
{
rotate(h);
rotate(h);
}
else
{
rotate(fa);
rotate(h);
}
}
}
void access(int x)
{
int s=0;

while(x)
{
splay(x);
num[x]+=siz[r[x]];
r[x]=s;
num[x]-=siz[r[x]];
push_up(x);
s=x;x=f[x];
}
}
void makeroot(int x)
{
access(x);
splay(x);
rev[x]^=1;
}
int findv(int u,long long v)
{
int st=0;
while(u)
{
push_down(u);
if((val[u]+sum[l[u]])*2<=v)
{
st=u;
v-=(val[u]+sum[l[u]])*2;
u=r[u];
}
else
u=l[u];
}
return st;
}//这个写法的find好像快一些
void dfs_bt(int u)
{
int v;
if(u<=n) siz[u]=1;
else siz[u]=0;
for(int i=beg[u];i;i=w[i][1])
{
v=w[i][0];
if(v==f[u])
continue;
f[v]=u;
dfs_bt(v);
siz[u]+=siz[v];
}
num[u]=siz[u];
}
void printh(int h)
{
push_down(h);
if(l[h]) printh(l[h]);
cerr<<h<<" ";
if(r[h]) printh(r[h]);
}
void printd(int h)
{
push_down(h);
if(l[h]) printd(l[h]);
if(r[h]) printd(r[h]);
}//调试用
}lct;
long long work(int s,int t,long long v)
{
if(s==t) return 0;
lct.makeroot(t);
lct.access(s);
lct.splay(s);
long long tt=lct.sum[s];
if(v>=tt) return 0;
int p=lct.findv(s,tt+v);
if(!p) return 0;
lct.splay(p);
lct.push_down(p);
lct.push_down(lct.r[p]);
long long cnt=lct.siz[lct.r[p]];
return cnt*(tt-v)-lct.ran[lct.r[p]]*2;
}
int main(){
//  freopen("in.txt","r",stdin);
//  freopen("out.txt","w",stdout);
int Q;
int s,t;
long long v;
long long ans1,ans2;
while(scanf("%d",&n)!=EOF)
{
lct.csh();
for(int i=1;i<=n-1;i++)
{
scanf("%d%d%lld",&s,&t,&v);
lct.val[n+i]=lct.sum[n+i]=v;
putin(s,n+i,v);
putin(n+i,s,v);
putin(t,n+i,v);
putin(n+i,t,v);//把边当做点
}
lct.dfs_bt(1);
scanf("%d",&Q);
while(Q--)
{
scanf("%d%d%lld",&s,&t,&v);
ans2=work(t,s,v);
ans1=work(s,t,v);
printf("%lld\n",ans1+ans2);

}
}
return 0;
}