FZU 2082(树链剖分模板题)

来源：互联网发布：去掉重复查数据库编辑：程序博客网时间：2024/06/05 12:07
一颗树边上的权在变动，动态求两点之间的和。
#include <iostream>#include <string.h>#include <math.h>#include <queue>#include <algorithm>#include <stdlib.h>#include <map>#include <set>#include <stdio.h>using namespace std;#define LL long long#define pi acos(-1.0)//#pragma comment(linker, "/STACK:1024000000")const int mod=1e9+7;const int INF=0x3f3f3f3f;const double eqs=1e-3;const int MAXN=50000+10;#define root 1, tot, 1#define lson l, mid, rt<<1#define rson mid+1, r, rt<<1|1int head[MAXN], cnt, tot;int siz[MAXN], top[MAXN], fa[MAXN], w[MAXN], son[MAXN], dep[MAXN];int sum[MAXN<<2];struct node{        int u, v, w, next;}edge[MAXN<<1];void add(int u,int v, int w){        edge[cnt].u=u;        edge[cnt].v=v;        edge[cnt].w=w;        edge[cnt].next=head[u];        head[u]=cnt++;}void init(){        memset(head,-1,sizeof(head));        cnt=tot=0;        memset(dep,0,sizeof(dep));        memset(son,0,sizeof(son));        memset(sum,0,sizeof(sum));}//第一次dfs任务完成统计//每棵子树大小siz//节点深度dep//重孩子(siz最大的孩子编号)son,//u初始值为树根，p为父节点void dfs1(int u, int p){        siz[u]=1;        for(int i=head[u];i!=-1;i=edge[i].next){                int v=edge[i].v;                if(v==p) continue ;                fa[v]=u;                dep[v]=dep[u]+1;                dfs1(v,u);                siz[u]+=siz[v];                if(siz[son[u]]<siz[v])son[u]=v;        }}//完成剖分,w为每个点在线段树种的位置,top为被剖分的这条链的顶端，//u初始值为树根，tp为线段树起点减去1 即为0void dfs2(int u, int tp){        w[u]=++tot; top[u]=tp;        if(son[u]) dfs2(son[u],tp);        for(int i=head[u];i!=-1;i=edge[i].next){                int v=edge[i].v;                if(v!=fa[u]&&v!=son[u])                        dfs2(v,v);        }}struct Line_Tree{        void PushUp(int rt)        {                sum[rt]=sum[rt<<1]+sum[rt<<1|1];        }        void Update(int p, int x, int l, int r, int rt)        {                if(l==r){                        sum[rt]=x;                        return ;                }                int mid=l+r>>1;                if(p<=mid) Update(p,x,lson);                else Update(p,x,rson);                PushUp(rt);        }        int Query(int ll, int rr, int l, int r, int rt)        {                if(ll<=l&&rr>=r){                        return sum[rt];                }                int mid=l+r>>1, ans=0;                if(ll<=mid) ans+=Query(ll,rr,lson);                if(rr>mid) ans+=Query(ll,rr,rson);                return ans;        }}lt;//从两个点相互爬升，完成对路径的分割//v -----> top[v] --> fa[top[v]]int solve(int u, int v){        int f1=top[u], f2=top[v], ans=0;        while(f1!=f2){                if(dep[f1]<dep[f2]){                        swap(u,v);                        swap(f1,f2);                }                ans += lt.Query(w[f1],w[u],root);                u=fa[f1];f1=top[u];        }        if(u==v) return ans;        if(dep[u]<dep[v]) swap(u,v);        return ans+lt.Query(w[son[v]],w[u],root);}/*说明一下点剖分 和边剖分的处理区别。两个dfs都一样，先把点剖分出来，如果需要用到边剖分，那么剖分出来的点看成该点与其父节点的连边，所以没更新一个边的值时取深度较大的点更新即可。*/int n,m;int main(){     while(scanf("%d %d",&n,&m)==2){          init();          for(int i =0 ; i<n - 1;i++){               int u , v , z;               scanf("%d %d %d",&u,&v,&z);               add(u , v ,z);               add(v , u ,z);          }          dfs1(1 , -1);          dfs2(1 , 0);          for(int i =0 ; i<cnt ; i+=2){               int u = edge[i].u , v= edge[i].v ;               if(dep[u] < dep[v]) swap(u , v);               lt.Update(w[u] , edge[i].w , root);          }          while(m--){               int cmd , x , y;               scanf("%d %d %d",&cmd , &x ,&y);               if(cmd == 0){                   node& e = edge[x * 2 - 1];                   int u = e.u , v = e.v;                   if(dep[u] < dep[v]) swap(u , v);                   lt.Update(w[u] , y , root);               }  else {                   printf("%d\n",solve(x , y));               }          }     }}
0 0