hdu_5293_Tree chain problem(DFS序+树形DP+LCA)

题目连接：http://acm.hdu.edu.cn/showproblem.php?pid=5293

被这题打蹦了，看着题解写的，很是爆炸，确实想不到，我用的DFS序+LCA+树形DP，当然也可以写树剖，不过这里DFS序更简单，因为都是对点到根的操作

 #include<cstdio>
 #include<vector>
 #include<algorithm>
 #pragma comment(linker, "/STACK:102400000,102400000")
 #define F(i,a,b) for(int i=a;i<=b;i++)
 #define root 1,n,1
 #define ls l,m,rt<<1
 #define rs m+1,r,rt<<1|1
 using namespace std;

 const int N=1e5+,DEG=;
 int t,n,m,x,y,dp[N],sum[N],c1[N*],c2[N*];
 struct edge{int u,v,w,lca;}p[N];
 vector<int>vec[N];
 inline void add(int x,int k,int *c){for(;x<=n*;x+=x&-x)c[x]+=k;}
 inline int ask(int x,int *c){int an=;while(x)an+=c[x],x-=x&-x;return an;}
 //LCA+dfs序
 int ed,g[N],nxt[*N],v[*N],fa[N][DEG],dep[N],idx,l[*N],r[*N];
 inline void adg(int x,int y){v[++ed]=y,nxt[ed]=g[x],g[x]=ed;}

 void LCA_dfs(int u,int pre){
     dep[u]=dep[pre]+,fa[u][]=pre,l[u]=++idx;
     F(i,,DEG-)fa[u][i]=fa[fa[u][i-]][i-];
     for(int i=g[u];~i;i=nxt[i])if(v[i]!=pre)LCA_dfs(v[i],u);
     r[u]=++idx;
 }

 int LCA(int a,int b){
     if(dep[a]>dep[b])a^=b,b^=a,a^=b;
     if(dep[a]<dep[b])F(i,,DEG-)if((dep[b]-dep[a])&(<<i))b=fa[b][i];
     if(a!=b)for(int i=DEG-;i<?a=fa[a][]:,i>=;i--)
     if(fa[a][i]!=fa[b][i])a=fa[a][i],b=fa[b][i];
     return a;
 }

 void fuck(int s,int fa){
     dp[s]=sum[s]=;
     for(int i=g[s];~i;i=nxt[i])if(v[i]!=fa)fuck(v[i],s),sum[s]+=dp[v[i]];
     dp[s]=sum[s];
     for(int i=;i<vec[s].size();i++){
         int uu=p[vec[s][i]].u,vv=p[vec[s][i]].v;
         int tmp=ask(l[uu],c1)+ask(l[vv],c1)-ask(l[uu],c2)-ask(l[vv],c2)+sum[s];
         dp[s]=max(dp[s],tmp+p[vec[s][i]].w);
     }
     add(l[s],sum[s],c1),add(r[s],-sum[s],c1);
     add(l[s],dp[s],c2),add(r[s],-dp[s],c2);
 }
 void init(){
     F(i,,n)g[i]=-,vec[i].clear();ed=idx=;
     F(i,,n*)c1[i]=c2[i]=;
 }

 int main(){
     scanf("%d",&t);
     while(t--){
         scanf("%d%d",&n,&m);
         init();
         F(i,,n-)scanf("%d%d",&x,&y),adg(x,y),adg(y,x);
         LCA_dfs(,);
         F(i,,m){
             scanf("%d%d%d",&p[i].u,&p[i].v,&p[i].w);
             p[i].lca=LCA(p[i].u,p[i].v);
             vec[p[i].lca].push_back(i);
         }
         fuck(,);
         printf("%d\n",dp[]);
     }
     return ;
 }