题意:抽象出来就是 一棵已知节点之间的边权,两个操作,1·修改边权,2·询问两个节点之间的边权和。
AC代码:
#include <string.h> #include <stdio.h> #include <algorithm> using namespace std; #define lson l,m,rt<<1 #define rson m+1,r,rt<<1|1 const int maxn=100000+10; const int inf=9999999999; struct Edge{ int to,next; }; struct Edge edge[maxn*2]; int head[maxn],tot; int stree[maxn];//节点v在线段树上的编号 int pre[maxn];//在线段树中编号为v节点所对应原图中的编号。 int fa[maxn];//f[v] 表示v的父亲节点 int son[maxn];//表示 与v同在一重链上的儿子节点(重儿子) int num[maxn];//num[v] v为根的子树的节点数。 int top[maxn];//top[v] v所在链的顶端节点。 int deep[maxn];//deep[v] 表示v的深度(根的深度为1) int data[maxn]; int pos,n; void addedge(int u,int v){ edge[tot].to=v; edge[tot].next=head[u]; head[u]=tot++; } void dfs1(int u,int father,int dep){ int i; deep[u]=dep; fa[u]=father; num[u]=1; for(i=head[u];i!=-1;i=edge[i].next){ int v=edge[i].to; if(v!=father){ dfs1(v,u,dep+1); num[u]+=num[v]; if(son[u]==-1 || num[v]>num[son[u]]){ son[u]=v; } } } } //给点编号得到线段树上的对应标号 void getpos(int u,int sp){ int i; top[u]=sp; stree[u]=pos++; pre[stree[u]]=u; if(son[u]==-1) return ; getpos(son[u],sp); for(i=head[u];i!=-1;i=edge[i].next){ int v=edge[i].to; if(v!=son[u] && v!=fa[u]){ getpos(v,v); } } } //树状数组or线段树or其他 int Max(int a,int b){ if(a>b) return a; return b; } struct node{ int l,r; int sum,max; }; struct node tree[maxn*3]; void PushUp(int rt){ tree[rt].sum=tree[rt<<1].sum+tree[rt<<1|1].sum; } void build(int l,int r,int rt){ tree[rt].l=l; tree[rt].r=r; tree[rt].sum=0; if(l==r){ return ; } int m=(tree[rt].l+tree[rt].r)>>1; build(lson); build(rson); PushUp(rt); } void update(int p,int add,int rt){ if(tree[rt].l==tree[rt].r && tree[rt].l==p){ tree[rt].sum=add; return ; } int m=(tree[rt].l+tree[rt].r)>>1; if(p<=m) update(p,add,rt<<1); else update(p,add,rt<<1|1); PushUp(rt); } int query_sum(int L,int R,int rt){ if(L<=tree[rt].l && tree[rt].r<=R){ return tree[rt].sum; } int m=(tree[rt].l+tree[rt].r)>>1; int ret=0; if(L<=m) ret+=query_sum(L,R,rt<<1); if(R>m) ret+=query_sum(L,R,rt<<1|1); return ret; } int find_sum(int u,int v){ int f1,f2,ans=0; f1=top[u]; f2=top[v]; while(f1!=f2){ //始终f1的深度比f2要大。深度大先更新 if(deep[f1]<deep[f2]){ swap(f1,f2); swap(u,v); } ans+=query_sum(stree[f1],stree[u],1); u=fa[f1]; f1=top[u]; } if(u==v) return ans; if(deep[u]>deep[v]) swap(u,v); ans+=query_sum(stree[son[u]],stree[v],1); return ans; } void init(){ tot=0; memset(head,-1,sizeof head); pos=0; memset(son,-1,sizeof son); } int e[maxn][5]; int main() { int t,x,y; int q,i,s,tmp; init(); scanf("%d%d%d",&n,&q,&s); tmp=s; for(i=0;i<n-1;i++){ scanf("%d %d %d",&e[i][0],&e[i][1],&e[i][2]); addedge(e[i][0],e[i][1]); addedge(e[i][1],e[i][0]); } dfs1(1,0,0);//根的deep是0.father从0开始。 getpos(1,1); build(1,n,1); for(i=0;i<n-1;i++){ if(deep[e[i][0]]>deep[e[i][1]]) swap(e[i][0],e[i][1]); update(stree[e[i][1]],e[i][2],1); } int op; while(q--){ int a,b; scanf("%d",&op); if(op==0){ scanf("%d",&a); printf("%d\n",find_sum(tmp,a)); tmp=a; } else{ scanf("%d %d",&a,&b); update(stree[e[a-1][1]],b,1); } } return 0; }