[bzoj1060][树形DP]时态同步

关注

原创最新推荐文章于 2020-10-26 20:15:49 发布 285 阅读 AI 写同款· GEO 优化 ›

AI权益加码！Claude Code、Cursor等20+工具免费用！购周边限时加赠Coding Plan Lite，畅享主流AI工具！学习进阶更高效！阅读详情

Description

　小Q在电子工艺实习课上学习焊接电路板。一块电路板由若干个元件组成，我们不妨称之为节点，并将其用数
字1,2,3….进行标号。电路板的各个节点由若干不相交的导线相连接，且对于电路板的任何两个节点，都存在且仅
存在一条通路（通路指连接两个元件的导线序列）。在电路板上存在一个特殊的元件称为“激发器”。当激发器工
作后，产生一个激励电流，通过导线传向每一个它所连接的节点。而中间节点接收到激励电流后，得到信息，并将
该激励电流传向与它连接并且尚未接收到激励电流的节点。最终，激烈电流将到达一些“终止节点”——接收激励
电流之后不再转发的节点。激励电流在导线上的传播是需要花费时间的，对于每条边e，激励电流通过它需要的时
间为te，而节点接收到激励电流后的转发可以认为是在瞬间完成的。现在这块电路板要求每一个“终止节点”同时
得到激励电路——即保持时态同步。由于当前的构造并不符合时态同步的要求，故需要通过改变连接线的构造。目
前小Q有一个道具，使用一次该道具，可以使得激励电流通过某条连接导线的时间增加一个单位。请问小Q最少使用
多少次道具才可使得所有的“终止节点”时态同步？

Input

　　第一行包含一个正整数N，表示电路板中节点的个数。第二行包含一个整数S，为该电路板的激发器的编号。接下来N-1行，每行三个整数a , b
, t。表示该条导线连接节点a与节点b，且激励电流通过这条导线需要t个单位时间

Output

　　仅包含一个整数V，为小Q最少使用的道具次数

Sample Input

3

1

1 2 1

1 3 3

Sample Output

2

HINT

N ≤ 500000，t ≤ 1000000

题解

f[i][0]表示第i个点的子树中叶子节点到i点最少使用多少次道具才能同步
f[i][1]表示第i个点的子树中叶子节点到i点最少使用道具情况下的最短距离
由于第i个点的子树是共享root~i的这条链的，所以假如要使叶子节点到根的距离相等，那么至少要满足到i点距离均为相等
合并的时候枚举儿子y，设当前的根为x
看一看y的距离是否比x的距离短，是的话就改x~y的那条边，否则改x处理过的子树那些边
因为y到x的影响是不能再改变的嘛。。
记得开longlong

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
struct node
{
    int x,y,next;LL c;
}a[1100000];int len,last[510000];
void ins(int x,int y,LL c)
{
    len++;
    a[len].x=x;a[len].y=y;a[len].c=c;
    a[len].next=last[x];last[x]=len;
}
LL f[510000][2];int n,S,G[510000];
void treedp(int x,int fa)
{
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=fa)
        {
            treedp(y,x);
            int sx=f[x][0],tx=f[x][1],sy=f[y][0],ty=f[y][1]+a[k].c;
            if(tx>ty)f[x][0]+=tx-ty+f[y][0];
            else
            {
                f[x][0]+=G[x]*(ty-tx)+f[y][0];
                f[x][1]=ty;
            }
            G[x]++;
        }
    }
}
int main()
{
    scanf("%d%d",&n,&S);
    len=0;memset(last,0,sizeof(last));
    for(int i=1;i<n;i++)
    {
        int x,y;LL c;
        scanf("%d%d%lld",&x,&y,&c);
        ins(x,y,c);ins(y,x,c);
    }
    memset(G,0,sizeof(G));
    treedp(S,0);
    printf("%lld\n",f[S][0]);
    return 0;
}