[树上差分]天天爱跑步

最新推荐文章于 2022-05-01 08:22:29 发布

Wall-E99

最新推荐文章于 2022-05-01 08:22:29 发布

阅读量175

点赞数

分类专栏：树上差分

本文链接：https://blog.csdn.net/zzrh2018/article/details/82025107

版权

树上差分专栏收录该内容

1 篇文章 0 订阅

订阅专栏

一道纠结了很久的题

重点不知道的是差分是对于每个桶而言的

树上差分能解决一部分树上统计问题

u表示起点，v表示终点

对于一条u到v的路径，先讨论LCA!=u&&LCA!=v的情况：

分为u到LCA的路径和LCA到v的路径

对于u到LCA的路径上的点x，当deep[u]-deep[x]=w[x]时，即w[x]+deep[x]=deep[u]时，这条路径对点x有贡献；

观察发现w[x]+deep[x]是定值，所以统计经过x的路径中，deep[u]=w[x]+deep[x]的路径条数。

对于LCA到v的路径上的点x，当deep[u]-2*deep[LCA]+deep[x]=w[x]时，即w[x]-deep[x]=deep[u]-2*deep[lca]时，这条路径对点x有贡献；

观察发现w[x]-deep[x]是定值，所以统计经过x的路径中，deep[u]-2*deep[lca]=w[x]-deep[x]的路径条数；

接下来就是统计路径条数了，用到树上差分

我们统计的起点(终点）一定在点x子树内，所以统计x子树内有多少起点（终点)的值等于所需值

即统计有多少个在点x子树内的起点的deep[u]的值与deep[x]+w[x]相同

有多少终点的deep[u]-2*deep[lca]与w[x]-deep[x]相同

对于一个值，再u、v上加一个表示这个值+1的标记

考虑到x子树内的路径不一定经过x，所以在father[LCA]上加一个标记表示这个值-1

标记用动态数组储存然后一遍dfs用两个桶分别统计，统计时值统一加上n，因为可能出现负数

记录下dfs到父亲节点时自己（也就是父亲的儿子）所需值的个数，然后统计完子树的值之后再做差计算自己

对于LCA==u||LCA==v的情况归于以上两类计算，特殊处理一下

另外，对于分裂成两条链LCA可能会被统计两遍，最后特殊判断一下，如果被统计了两遍就减去一遍,

复杂度：LCA O(mlogn）dfs统计 O(n)

#include<bits/stdc++.h>
using namespace std;
const int N=350000;
int Max,x,y,z,len,n,m,tot,w[N],tong1[4*N],tong2[4*N],ans[N],to[N*2],head[N],next1[N*2],d[N],f[N][30];
struct p{
    int type,zhi,qz;
};
int read(){
    char ch=getchar();int x=0,op=1;
    for (;!isdigit(ch);ch=getchar()) if (ch=='-') op=-1;
    for (;isdigit(ch);ch=getchar()) x=(x<<1)+(x<<3)+ch-'0';
    return x*op;
}
vector <p> t[4*N];
void add(int x,int y)
{  next1[++tot]=head[x];
   head[x]=tot;
   to[tot]=y;
}
void dfs(int x)
{  for (int i=head[x];i;i=next1[i])
    {  int v=to[i]; 
       if (!d[v]) 
       { d[v]=d[x]+1; 
         f[v][0]=x; 
         dfs(v);}
    }
}
int  lca(int x,int y)
{    if (d[x]<d[y])  swap(x,y);
   int tmp=d[x]-d[y];
   for (int i=25;i>=0;i--)  
   if ((tmp>>i)&1) 
    {x=f[x][i];}
   if (x==y) { return x;}
   else {for (int i=25;i>=0;i--)  
         if (f[x][i]!=f[y][i]) { x=f[x][i]; y=f[y][i];}
         x=f[x][0]; return x;
        }
}
void solve(int x)
{   int tmp=tong1[d[x]+w[x]]+tong2[d[x]-w[x]+Max];//tong1记录向上走的链 tong2记录向下走的链 
    for (int i=0;i<t[x].size();i++)
    { if (t[x][i].type==1)  {tong1[t[x][i].zhi]+=t[x][i].qz;}
                       else {tong2[t[x][i].zhi]+=t[x][i].qz;}}
   for (int i=head[x];i;i=next1[i])
   { if (d[to[i]]>d[x]) { solve(to[i]);} }
   ans[x]=tong1[d[x]+w[x]]+tong2[d[x]-w[x]+Max]-tmp;
}
int main()
{  n=read(); m=read();  Max=3*n;
   for (int i=1;i<=n-1;i++)
   {  x=read(); y=read();  add(x,y); add(y,x);}
     d[1]=1;dfs(1);
  	 for (int i=1;i<=n;i++)
       for (int j=1;(1<<j)<d[i];j++)
       {  f[i][j]=f[f[i][j-1]][j-1];}
   for (int i=1;i<=n;i++)   w[i]=read();
   for (int i=1;i<=m;i++)
   {  x=read(); y=read(); z=lca(x,y);
       len=d[x]+d[y]-2*d[z];
       t[x].push_back(p{1,d[x],1});
       t[y].push_back(p{2,d[y]-len+Max,1});
       t[z].push_back(p{1,d[x],-1});
       t[f[z][0]].push_back(p{2,d[y]-len+Max,-1});//对桶进行树上差分
   }
   solve(1);
   for (int i=1;i<=n-1;i++)  cout<<ans[i]<<' ';
   cout<<ans[n]<<endl;	 	 
}