题意:给定一个树形图,某人原来在 s 点,每条边(路)有通过的时间花费,有两种操作:1. 查询某人到 u 点花费的时间 2. 更新某条路的时间花费。
解题思路:前面树链剖分练得一直是点权,但一遇边权懵逼了。下里面的代码是从点权改过来的。其实边权和点权是差不多的。大致意思就是,取每条边连个点中,处于下位的(更深)点来记录这条边的值。有一些查询时的细节和点权稍有不同,详见代码注释。
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#pragma comment(linker, "/STACK:102400000,102400000")
#define N 100050
using namespace std;
struct node
{
int v,next;
}s[2*N];
struct edge
{
int u,v,c;
}e[N];
struct tree
{
int l,r,w,c;
}p[3*N];
int w[N],head[N],cnt,fa[N],siz[N],son[N],dep[N],top[N],id[N];
void add_edge(int u,int v)
{
s[cnt].v=v;
s[cnt].next=head[u];
head[u]=cnt++;
s[cnt].v=u;
s[cnt].next=head[v];
head[v]=cnt++;
}
void dfs1(int u,int f,int deep)
{
dep[u]=deep;//标记每个点的深度
fa[u]=f;//标记每个点的父亲
siz[u]=1;//标记每个非叶子节点的子树大小
int maxson=-1;
for(int i=head[u];i!=-1;i=s[i].next)
{
int v=s[i].v;
if(v==f) continue;
dfs1(v,u,deep+1);
siz[u]+=siz[v];
if(siz[v]>maxson)
{
son[u]=v;
maxson=siz[v];
}
}
}
void dfs2(int u,int t)
{
id[u]=++cnt;
top[u]=t;
if(son[u]==-1) return;
dfs2(son[u],t);
for(int i=head[u];i!=-1;i=s[i].next)
{
int v=s[i].v;
if(v==fa[u]||v==son[u]) continue;
dfs2(v,v);
}
}
void build(int rt,int l,int r)
{
p[rt].l=l,p[rt].r=r,p[rt].c=0,p[rt].w=0;
if(l==r) return;
int mid=(l+r)/2;
build(2*rt,l,mid);
build(2*rt+1,mid+1,r);
}
int query(int rt,int l,int r)
{
if(l==p[rt].l&&r==p[rt].r)
{
return p[rt].w;
}
int mid=(p[rt].l+p[rt].r)/2;
if(r<=mid)
{
return query(2*rt,l,r);
}else
if(l>=mid+1)
{
return query(2*rt+1,l,r);
}else
{
return (query(2*rt+1,mid+1,r)+query(2*rt,l,mid));
}
}
void update(int rt,int l,int r,int kk)
{
if(l==p[rt].l&&r==p[rt].r)
{
p[rt].w=kk;
return;
}
int mid=(p[rt].l+p[rt].r)/2;
if(r<=mid)
{
update(2*rt,l,r,kk);
}else
if(l>=mid+1)
{
update(2*rt+1,l,r,kk);
}else
{
update(2*rt+1,mid+1,r,kk);
update(2*rt,l,mid,kk);
}
p[rt].w=(p[2*rt].w+p[2*rt+1].w);
}
long long qrange(int u,int v)
{
long long res=0;
while(top[u]!=top[v])
{
if(dep[top[u]]<dep[top[v]]) swap(u,v);
res+=query(1,id[top[u]],id[u]);
u=fa[top[u]];
}
if(u==v) return res;//必须加上这句话,因为下面要用到son[u],万一u==v的话就永远也找不到,会runtime error
if(dep[u]>dep[v]) swap(u,v);
res+=query(1,id[son[u]],id[v]);//最后汇合点的上位边是不能算入的,这点一定要注意
return res;
}
void uprange(int u,int v,int k)
{
while(top[u]!=top[v])
{
if(dep[top[u]]<dep[top[v]]) swap(u,v);
update(1,id[top[u]],id[u],k);
u=fa[top[u]];
}
if(dep[u]>dep[v]) swap(u,v);
update(1,id[u],id[v],k);
}
int main()
{
//freopen("t.txt","r",stdin);
int n,m,a,b,q,tp,st;
scanf("%d%d%d",&n,&q,&st);
cnt=0;
memset(fa,-1,sizeof(fa));
memset(head,-1,sizeof(head));
memset(son,-1,sizeof(son));
for(int i=1;i<n;i++)
{
scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].c);
add_edge(e[i].u,e[i].v);
}
dfs1(1,-1,1);
cnt=0;
dfs2(1,1);
build(1,1,n);
for(int i=1;i<n;i++)
if(dep[e[i].u]>dep[e[i].v]) uprange(e[i].u,e[i].u,e[i].c);
else uprange(e[i].v,e[i].v,e[i].c);
while(q--)
{
scanf("%d",&tp);
if(tp==1)
{
scanf("%d%d",&a,&b);
if(dep[e[a].u]>dep[e[a].v]) uprange(e[a].u,e[a].u,b);
else uprange(e[a].v,e[a].v,b);
}else
{
scanf("%d",&a);
printf("%d\n",qrange(st,a));
st=a;
}
}
return 0;
}