codechef FUN WITH TREES

题目大意：

给一棵树root=1的树：

给一些操作：u  v 的路径所有节点的node + val；

最后m个询问：u 节点（包括u） sum%mod 是多少。

LCA + RMQ；

我们每次mark ,u , v, +=val;

fa=lca[u,v])-=val;

fa[fa]-=val

你会发现下次DFS一边的时候可以把所以答案更新出来。DFS 从下往上更新。

 #include <stdio.h>
 #include <string.h>
 #include <iostream>
 #include <algorithm>
 #include <vector>
 #include <queue>
 #include <set>
 #include <map>
 #include <string>
 #include <math.h>
 #include <stdlib.h>
 #include <time.h>
 using namespace std;
 #define N 100001
 #define LN 20
 #define mod 1000000007

 vector<int>mp[N];
 int mark[N];
 int dp[LN][N];
 int dep[N];
 int ans[N],tmp[N];

 int lca(int l,int r)
 {
   if (dep[l]<dep[r]) swap(l,r);
   int dif=dep[l]-dep[r];
   for (int i=;i<LN;i++)
   if ((dif>>i)&) l=dp[i][l];

   if (l==r)  return l;

   for (int i=LN-;i>=;i--)
   if (dp[i][l]!=dp[i][r]) l=dp[i][l],r=dp[i][r];
   return dp[][l];
 }

 void dfs(int u,int fa,int level)
 {
    dp[][u]=fa;
    dep[u]=level;
    for (int i=;i<mp[u].size();i++)
    {
      int  v=mp[u][i];
      if (v==fa) continue;
      dfs(v,u,level+);
    }
 }

 int dfs2(int u,int fa=)
 {
   int sum=mark[u];
   for (int i=;i<mp[u].size();i++)
   {
     int v=mp[u][i];
     if (v==fa) continue;
     dfs2(v,u);
     sum+=tmp[v];
     sum%=mod;
     ans[u]+=ans[v];
     ans[u]%=mod;
   }
   tmp[u]=sum;
   ans[u]+=tmp[u];
   ans[u]%=mod;
 }

 int main()
 {
    int n,u,q;
    scanf("%d",&n);
    scanf("%d%d",&u,&q);
   // for (int i=0;i<=n;i++) mp[i].clear();
    for (int i=;i<n;i++)
    {
      int a,b;
      scanf("%d%d",&a,&b);
      mp[a].push_back(b);
      mp[b].push_back(a);
    }
    dfs(,,);
    for (int j=;j<LN;j++)
    for (int i=;i<=n;i++)
    dp[j][i]=dp[j-][dp[j-][i]];
    while (u--)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        mark[a]+=c;
        mark[b]+=c;
        mark[a]%=mod;
        mark[b]%=mod;
        int l=lca(a,b);
        mark[l]-=c;
        mark[l]%=mod;
        mark[dp[][l]]-=c;
        mark[dp[][l]]%=mod;
    }

    dfs2();
    while (q--)
    {
      int x;
      scanf("%d",&x);
      printf("%d\n",(ans[x]+mod)%mod);
    }

    return ;
 }