BZOJ 1060 [ZJOI2007]时态同步

题目链接：传送门
题面：
$$1060:[ZJOI2007]时态同步$$

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，te ≤ 1000000

听说是树形dp
可是这么好写的暴力怎么忍心写dp呢
从它给定的节点开始$dfs$
先搜到叶节点
因为要从叶节点往上更新
然后记录路径上最大的边权
我们要把这个路径上的边都改成这么大
再扫一遍统计答案
最后要更新父节点的边权
就是改成子节点（严格来说是子树）的最大边权加上它的子节点到它的边权
树形dp做法的话
博主不会啊！
参考其他博客即可
至此。
哦记得开long long

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <complex>
#include <algorithm>
#include <climits>
#include <queue>
#include <map>
#include <vector>
#include <iomanip>
#define A 1000010
#define B 2010
#define ll long long

using namespace std;
struct node {
    int next, to, dis;
}edge[A];
int head[A], num_edge;
void add_edge(int from, int to, int dis) {
    edge[++num_edge].next = head[from];
    edge[num_edge].to = to;
    edge[num_edge].dis = dis;
    head[from] = num_edge;
}
int n, s, a, b, c;
ll ans;
void dfs(int fr, int fa) {
    for (int i = head[fr]; i; i = edge[i].next) {
        int ca = edge[i].to;
        if (ca == fa) continue;
        dfs(ca, fr);
    }
    int maxx = 0;
    for (int i = head[fr]; i; i = edge[i].next) {
        int ca = edge[i].to;
        if (ca == fa) continue;
        maxx = max(maxx, edge[i].dis);
    }
    for (int i = head[fr]; i; i = edge[i].next) {
        int ca = edge[i].to;
        if (ca == fa) continue;
        ans += maxx - edge[i].dis;
    }
    for (int i = head[fa]; i; i = edge[i].next) {
        int ca = edge[i].to;
        if (ca != fr) continue;
        edge[i].dis += maxx;
        break;
    }
}

int main() {
    cin >> n >> s;
    for (int i = 1; i < n; i++) {
        cin >> a >> b >> c;
        add_edge(a, b, c);
        add_edge(b, a, c);
    }
    dfs(s, 0);
    cout << ans << endl;
}