题意:
给你n个点的树,每个边有个权值,开始处于点m,现在有q次操作,每次单点修改或者将点移动到u,对于每次移动输出经过路程的权值
思路:
树链剖分的基础题,不同点在于每次记得更新点时候把点的更新即可
错误及反思:
C++跑了1700,G++跑了2200,hdu上faq说过G++处理大量输入的时候会
代码:
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define fi first
#define se second
const int N =100010;
int segtree[N*4];
int top[N],si[N],fa[N],fi[N],son[N],depth[N],id[N],val[N];
pair<int,int> e[N*2],b[N];
int m,tot=0,tid=0,n,q;
void addedge(int x,int y)
{
e[tot].fi=y;
e[tot].se=fi[x];
fi[x]=tot++;
e[tot].fi=x;
e[tot].se=fi[y];
fi[y]=tot++;
}
void dfs1(int now,int bef,int dep)
{
fa[now]=bef;
depth[now]=dep;
si[now]=1;
for(int i=fi[now];i!=-1;i=e[i].se)
if(e[i].fi!=bef)
{
dfs1(e[i].fi,now,dep+1);
si[now]+=si[e[i].fi];
if(son[now]==-1) son[now]=e[i].fi;
else son[now]=si[e[i].fi]>si[son[now]]?e[i].fi:son[now];
}
}
void dfs2(int now,int tp)
{
top[now]=tp;
id[now]=tid++;
if(son[now]!=-1) dfs2(son[now],tp);
for(int i=fi[now];i!=-1;i=e[i].se)
if(e[i].fi!=fa[now]&&e[i].fi!=son[now])
dfs2(e[i].fi,e[i].fi);
}
void change(int pos,int v,int l,int r,int rt)
{
if(pos==l&&l==r)
{
segtree[rt]=v;
return;
}
int m=(l+r)/2;
if(m>=pos) change(pos,v,lson);
else change(pos,v,rson);
segtree[rt]=segtree[rt<<1]+segtree[rt<<1|1];
}
int query(int L,int R,int l,int r,int rt)
{
if(L<=l&&R>=r)
return segtree[rt];
int m=(l+r)/2;
int ans=0;
if(m>=L) ans+=query(L,R,lson);
if(m<R) ans+=query(L,R,rson);
return ans;
}
int modify(int L,int R)
{
int f1=top[L],f2=top[R];
int ans=0;
while(f1!=f2)
{
if(depth[f1]<depth[f2])
{
swap(f1,f2);
swap(L,R);
}
ans+=query(id[f1],id[L],0,tid-1,1);
L=fa[f1];
f1=top[L];
}
if(L==R) return ans;
if(depth[L]>depth[R]) swap(L,R);
ans+=query(id[son[L]],id[R],0,tid-1,1);
return ans;
}
int main()
{
memset(first,-1,sizeof(first));
memset(son,-1,sizeof(son));
scanf("%d%d%d",&n,&q,&m);
for(int i=0;i<n-1;i++)
{
scanf("%d%d%d",&b[i].fi,&b[i].se,&val[i]);
addedge(b[i].fi,b[i].se);
}
dfs1(1,1,1);
dfs2(1,1);
for(int i=0;i<n-1;i++)
{
if(depth[b[i].fi]<depth[b[i].se])
swap(b[i].fi,b[i].se);
change(id[b[i].fi],val[i],0,tid-1,1);
}
while(q--)
{
int ta,tb,tc;
scanf("%d",&ta);
if(ta)
{
scanf("%d%d",&tb,&tc);
change(id[b[tb-1].fi],tc,0,tid-1,1);
}
else
{
scanf("%d",&tb);
printf("%d\n",modify(m,tb));
m=tb;
}
}
}