Luogu P1131 [ZJOI2007]时态同步

最新推荐文章于 2024-04-20 12:02:00 发布

weixin_30911809

最新推荐文章于 2024-04-20 12:02:00 发布

阅读量290

点赞数

原文链接：http://www.cnblogs.com/NLDQY/p/11482647.html

版权

本文介绍了一种使用树形动态规划方法解决特定树结构中叶子节点到根节点距离相等问题的算法。通过定义状态转移方程，文章详细解释了如何找到最小边权增量，使所有叶子节点到根节点的距离相等。最后，提供了完整的代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：Click here

Solution:

题目描述十分冗长，实际上就是给你一棵树，问你最少增加多少次边权，使得所有叶子节点到根节点距离相等

有一个十分显然的结论：最后的距离为不增加边权的最大距离

则我们可设\(f[x]\)表示\(x\)节点与\(fa[x]\)相连的边需要加多少权值

易得状态转移：\(f[x]=min\{ f[v]|v\in son_x \}\)

最后对每个点统计答案：\(ans+=f[x]-f[fa[x]]\)

要开long long

Code:

//P1131 Treedp
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=5e5+1;
const int inf=21474836470000;
int n,rt,cnt,head[N];
int ans,maxdis,dis[N],f[N];
struct Edge{int nxt,to,val;}edge[N<<1];
void ins(int x,int y,int z){
    edge[++cnt].nxt=head[x];
    edge[cnt].to=y;head[x]=cnt;
    edge[cnt].val=z;
}
void getdis(int x,int fa){
    for(int i=head[x];i;i=edge[i].nxt){
        int y=edge[i].to;
        if(y==fa) continue;
        dis[y]=dis[x]+edge[i].val;
        getdis(y,x);maxdis=max(maxdis,dis[y]);
    }
}
void dfs(int x,int fa){
    f[x]=inf;int flag=0;
    for(int i=head[x];i;i=edge[i].nxt){
        int y=edge[i].to;
        if(y==fa) continue;
        dfs(y,x);f[x]=min(f[x],f[y]);
        flag=1;
    }
    if(!flag) f[x]=maxdis-dis[x];
}
void calc(int x,int fa){
    ans+=f[x]-f[fa];
    for(int i=head[x];i;i=edge[i].nxt){
        int y=edge[i].to;
        if(y==fa) continue;
        calc(y,x);  
    }
}
int read(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-f;ch=getchar();}
    while(isdigit(ch)){x=x*10+ch-48;ch=getchar();}
    return x*f;
}
signed main(){
    n=read(),rt=read();
    for(int i=1;i<n;i++){
        int x=read(),y=read(),z=read();
        ins(x,y,z),ins(y,x,z);
    }getdis(rt,0);
    dfs(rt,0);calc(rt,0);
    printf("%lld\n",ans);
    return 0;
}

转载于:https://www.cnblogs.com/NLDQY/p/11482647.html