poj-openjudge 1042:Moles 解题报告
题目
2013北大校赛
题意：
给出一棵带权树，树根为1，mole当前位于1，有三种操作
1 u：mole移到点u，并输出移动的距离
2：输出mole所在点到其子孙的最大距离
3 u v w：将一棵树边(u,v)的权值改为w
思路：
一开始想的是树链剖分，但是不会处理2，就算了
只好用挫一点的办法了，所以代码巨长......
对于2，可以先求得树的dfs序列和每个点到根1的距离，建立线段树，区间查询最大值。（比如样例中树的dfs序列是1 2 5 3 4，线段树中叶子节点的值为0 2 1 -3 7，则若要查询以结点2为根的子树，则相当于查询区间[2,3]的最大值）；
这样操作3也很容易想到，因为(u,v)一定是原来树上的边，不妨设u是父亲，v是儿子，那么更新边(u,v)，相当于以v为根的子树上所有点到根1的距离都要被更新，即对应的线段树区间更新。
考虑操作1，找到u和mole原来的位置v的公共祖先r，他们两之间的距离就是dis[u]+dis[v]-2*dis[r]
感觉对于每一个操作都可以出一题......
代码：
//time: 400ms
//memory: 7912KB
//length :4374B
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define maxn 100005
#define maxq 1000005
int n;
struct node
{
int v,w,next;
}e[maxn<<1],Q[maxq<<1];
int ecnt,pre[maxn],qcnt,prq[maxn],f[maxn];
bool vis[maxn];
int cnt,len[maxn],a[maxn],b[maxn],dis[maxn];
struct NODE
{
int a,b,c,d;
}qur[maxq];
int tree[maxn<<2],lazy[maxn<<2];
void adde(int u,int v,int w)
{
e[ecnt].v=v;
e[ecnt].w=w;
e[ecnt].next=pre[u];
pre[u]=ecnt++;
}
void addq(int u,int v,int w)
{
Q[qcnt].v=v;
Q[qcnt].w=w;
Q[qcnt].next=prq[u];
prq[u]=qcnt++;
}
void dfs(int u,int p)
{
a[u]=++cnt;
dis[cnt]=len[u];
for (int i=pre[u];i!=-1;i=e[i].next)
{
int v=e[i].v;
if (v==p) continue;
len[v]=len[u]+e[i].w;
dfs(v,u);
}
b[u]=cnt;
}
int find(int x)
{
if (f[x]==x) return x;
f[x]=find(f[x]);
return f[x];
}
void tarjan(int u,int p)
{
for (int i=pre[u];i!=-1;i=e[i].next)
{
int v=e[i].v;
if (v==p) continue;
tarjan(v,u);
f[v]=u;
}
vis[u]=1;
for (int i=prq[u];i!=-1;i=Q[i].next)
{
int v=Q[i].v;
if (vis[v]) qur[Q[i].w].c=find(v);
}
}
void build(int p,int l,int r)
{
lazy[p]=0;
if (l==r)
{
tree[p]=dis[l];
return;
}
int mid=(l+r)>>1;
build(p<<1,l,mid);
build(p<<1|1,mid+1,r);
tree[p]=max(tree[p<<1],tree[p<<1|1]);
}
void down(int p,int la)
{
if (la)
{
lazy[p<<1]+=la;
lazy[p<<1|1]+=la;
tree[p<<1]+=la;
tree[p<<1|1]+=la;
lazy[p]=0;
}
}
void insert(int p,int L,int R,int l,int r,int w)
{
if (l==L&&r==R)
{
tree[p]+=w;
lazy[p]+=w;
return;
}
down(p,lazy[p]);
int mid=(L+R)>>1;
if (r<=mid) insert(p<<1,L,mid,l,r,w);
else if (l>mid) insert(p<<1|1,mid+1,R,l,r,w);
else
{
insert(p<<1,L,mid,l,mid,w);
insert(p<<1|1,mid+1,R,mid+1,r,w);
}
tree[p]=max(tree[p<<1],tree[p<<1|1]);
}
int cal(int p,int L,int R,int l,int r)
{
if (l==L&&r==R)
return tree[p];
down(p,lazy[p]);
int mid=(L+R)>>1,res;
if (r<=mid) res=cal(p<<1,L,mid,l,r);
else if (l>mid) res=cal(p<<1|1,mid+1,R,l,r);
else res=max(cal(p<<1,L,mid,l,mid),cal(p<<1|1,mid+1,R,mid+1,r));
tree[p]=max(tree[p<<1],tree[p<<1|1]);
return res;
}
int main()
{
int cas,u,v,w,q;
scanf("%d",&cas);
while (cas--)
{
scanf("%d",&n);
ecnt=cnt=0;
memset(pre,-1,sizeof(pre));
for (int i=1;i<n;i++)
{
scanf("%d%d%d",&u,&v,&w);
adde(u,v,w);
adde(v,u,w);
}
dfs(1,0);
int root=1;
scanf("%d",&q);
qcnt=0;
memset(prq,-1,sizeof(prq));
for (int i=1;i<=q;i++)
{
scanf("%d",&w);
qur[i].d=w;
if (w==1)
{
scanf("%d",&v);
addq(root,v,i);
addq(v,root,i);
qur[i].a=v;
qur[i].b=root;
root=v;
}
else if (w==2)
qur[i].a=root;
else
{
scanf("%d%d%d",&u,&v,&w);
qur[i].a=u;
qur[i].b=v;
qur[i].c=w;
}
}
memset(vis,0,sizeof(vis));
for (int i=1;i<=n;i++) f[i]=i;
tarjan(1,0);
int ans;
build(1,1,n);
for (int i=1;i<=q;i++)
{
u=qur[i].a; v=qur[i].b; w=qur[i].c;
if (qur[i].d==1)
{
ans=cal(1,1,n,a[u],a[u]);
ans+=cal(1,1,n,a[v],a[v]);
ans-=2*cal(1,1,n,a[w],a[w]);
printf("%d\n",ans);
}
else if (qur[i].d==2)
{
if (a[u]==b[u]) ans=0;
else
{
ans=cal(1,1,n,a[u]+1,b[u]);
ans-=cal(1,1,n,a[u],a[u]);
}
printf("%d\n",max(0,ans));
}
else
{
if (a[u]>a[v]) swap(u,v);
int tmp=cal(1,1,n,a[v],a[v])-cal(1,1,n,a[u],a[u]);
w-=tmp;
insert(1,1,n,a[v],b[v],w);
}
}
}
return 0;
}