树上差分:点差分(P3128 [USACO15DEC]最大流Max Flow)

吐槽:

昨天学习了差分数组感觉很不错,通过差分递推答案,很节省时间,昨天晚上睡觉前学习了一波树上差分,没学懂,今天临走前补上,挺好

树上差分点差分:

我们要对树上一条链的权值进行修改,我们可以利用树上差分,比一个一个点修改快很多,如果一个一个点修改dfs()时间复杂度爆炸啊!

树上如何进行点差分:

先把两端点加上权值,然后把lca和它的父亲的权值减去,dfs一遍就把一条链上的权值给修改了

P3128 [USACO15DEC]最大流Max Flow
注意:因为有两个dfs()所以两个名字不要重,不然报错也找不到

AC代码:

#include<bits/stdc++.h>
using namespace std ;
//树上差分求最大点的权值
const int maxn=1e5+5;
struct node
{
    int to,nex;

```java
} tr[maxn*4];
int pre[maxn][32],depth[maxn],head[maxn];
int point[maxn];
//树上倍增lca
int n,m,cnt,ans;
void add(int u,int v)
{
    tr[cnt].to=v;
    tr[cnt].nex=head[u];
    head[u]=cnt++;
}
void dfs(int u,int fa)//求深度
{
    depth[u]=depth[fa]+1;
    pre[u][0]=fa;
    for(int i=1; (1<<i)<=depth[u]; i++)
        pre[u][i]=pre[pre[u][i-1]][i-1];
    for(int i=head[u]; ~i; i=tr[i].nex)
    {
        int v=tr[i].to;
        if(fa!=v)
        {
            dfs(v,u);
        }
    }
}
int lca(int u,int v)
{
    if(depth[u]<depth[v])
        swap(u,v);//让u的深度大
    int i=-1,j;
    while((1<<(i+1))<=depth[u])//从u跨到根节点的跨度
        i++;
    for(j=i; j>=0; j--) //同一水平
    {
        if(depth[u]-(1<<j)>=depth[v])
        {
            u=pre[u][j];
        }
    }
    if(u==v)
    {
        return u;
    }
    for(j=i; j>=0; j--)
    {
        if(pre[u][j]!=pre[v][j])
        {
            u=pre[u][j];
            v=pre[v][j];
        }
    }
    return pre[u][0];
}
//树上差分
void dfs_(int u,int fa)
{ //printf("%d\n",point[u]);
    for(int i=head[u];~i;i=tr[i].nex)
    {
        int v=tr[i].to;
        if(v==fa)
            continue;
         dfs_(v,u);
         point[u]+=point[v];
    }
    ans=max(ans,point[u]);
}
int main()
{
    scanf("%d %d",&n,&m);
    memset(head,-1,sizeof(head));
    int u,v;
    for(int i=1; i<n; i++)
    {
        scanf("%d %d",&u,&v);
        add(u,v);
        add(v,u);
    }
    dfs(1,0);
    for(int i=1;i<=m;i++)
    {
        scanf("%d %d",&u,&v);
        int lc=lca(u,v);
        //printf("%d\n",lc);
        point[u]++;
        point[v]++;
        point[lc]--;
        point[pre[lc][0]]--;

    }
    dfs_(1,0);
    printf("%d\n",ans);

}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值