【计蒜客习题】蒜头君的建设方案

---

 

说是树状DP的入门水题，然而甚至感觉不到DP的存在。。。

好吧，理论上这个确实是DP，由之前阶段推出当前阶段，以子结点为根的子树结点个数可以推出以父结点为根的子树结点个数。

然后针对每条边，根据题目要求的那样统计答案即可。不过我们统计时利用的应该是子结点的信息，怎么找到子结点？因为父结点的信息是由子结点的信息得来的，所以一定大于子结点。

另外，因为树状DP是建立在树这一数据结构上的，因此可以用到与树相关的各种特殊性质。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
inline int abs(int i) {if(i<0) i=-i;return i;}
const int maxn=1e6+5;
int n,head[maxn],eid,vis[maxn],size[maxn];
long long ans;
struct edge {
    int u,v,w,next;
    edge(int u=0,int v=0,int w=-1):u(u),v(v),w(w) {}
} E[2*maxn];
void insert(int u,int v,int w) {
    E[eid]=edge(u,v,w);
    E[eid].next=head[u];
    head[u]=eid++;
}
int dfs(int u) {
    size[u]=1;
    for(int p=head[u];p+1;p=E[p].next) {
        int v=E[p].v;
        if(vis[v]) continue;
        vis[v]=1;
        size[u]+=dfs(v);
    }
    return size[u];
}
int main() {
    scanf("%d",&n);
    memset(head,-1,sizeof(head));
    int s,t,c;
    for(int i=1;i<n;++i) {
        scanf("%d%d%d",&s,&t,&c);
        insert(s,t,c);
        insert(t,s,c);
    }
    vis[1]=1;
    dfs(1);
    for(int i=0;i<eid;i+=2) {
        int u=E[i].u,v=E[i].v,w=E[i].w;
        if(size[u]>size[v]) swap(u,v);
        ans+=w*1LL*abs(2*size[u]-n);
    }
    printf("%lld",ans);
    return 0;
}