bzoj 1060（Luogu 1131）（树形dp）

传送门
大水题，假的dp，差不多就是个贪心就完了。
求出p的子树到叶节点的最大距离，再对于每个儿子减去不够的那段（max_dis[pos]-edge[i].w-max_dis[son]）
没有什么要注意的，考试遇到这种一定要稳过。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=5e5+2;
int n,source,md[MAXN];
int head[MAXN],edge=0;
struct EDGE {
    int v,nxt,w;
}e[MAXN<<1];
ll ans=0;
inline int read() {
    int x=0,f=1;char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}
    while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x*f;
}
inline void adde(int u,int v,int w) {
    e[edge].nxt=head[u],e[edge].v=v,e[edge].w=w,head[u]=edge++;
    e[edge].nxt=head[v],e[edge].v=u,e[edge].w=w,head[v]=edge++;
}
void dfs(int p,int f) {
    for (int i=head[p];~i;i=e[i].nxt) {
        int v=e[i].v;
        if (v^f) {
            dfs(v,p);
            md[p]=max(md[p],e[i].w+md[v]);
        }
    }
    for (int i=head[p];~i;i=e[i].nxt) {
        int v=e[i].v;
        if (v^f)
            ans+=md[p]-e[i].w-md[v];
    }
}
int main() {
    memset(head,-1,sizeof(head));
    memset(md,0,sizeof(md));
    n=read(),source=read();
    for (register int i=1;i<n;++i) {
        int u=read(),v=read(),w=read();
        adde(u,v,w);
    }
    dfs(source,0);
    printf("%lld\n",ans);
    return 0;
}