ACDream dp专题 C - 哗啦啦村的扩建

中文题反而更容易让人误解，，所谓每条路的左边x个房子右边y个房子。。其实就是每条路的一端的房子数和另一端的房子数。然后按题目的公式算。

树形dp一个dfs就可以解决了， 求出每条路的一端连上了多少个房子， 另一端显然就是n减掉这一端的，坑就坑在题意了吧，，题目不难。

当然坑点也是有的，，万万分之一也就是说结果要乘1e8， 可是结果是0的时候又只能输出0。

#include <set>
#include <map>
#include <queue>
#include <stack>
#include <cmath>
#include <string>
#include <cctype>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <sstream>
#include <iostream>
#include <algorithm>
using namespace std;
#define ls id<<1,l,mid
#define rs id<<1|1,mid+1,r
#define OFF(x) memset(x,-1,sizeof x)
#define CLR(x) memset(x,0,sizeof x)
#define MEM(x) memset(x,0x3f,sizeof x)
typedef long long ll ;
typedef pair<int,int> pii ;
const int maxn = 1e5+50 ;
const int inf = 0x3f3f3f3f ;
const int MOD = 1e9+7 ;
int n,cnt;
int pnt[maxn],val[maxn],num[maxn];
int head[maxn],nxt[maxn];

void add_edge(int u,int v,int w) {
    pnt[cnt]=v;
    val[cnt]=w;
    nxt[cnt]=head[u];
    head[u]=cnt++;
}

int dfs(int u,int last) {
    int res=1;
    for (int i=head[u];~i;i=nxt[i]) {
        int v=pnt[i];
        if (v==last) continue ;
        num[i]=num[i^1]=dfs(v,u);
        res+=num[i];
    }
    return res;
}
int main () {
#ifdef LOCAL
	freopen("C:\\Users\\Administrator\\Desktop\\in.txt","r",stdin);
//      freopen("C:\\Users\\Administrator\\Desktop\\out.txt","w",stdout);
#endif
    while (~scanf("%d",&n)) {
        OFF(head);
        int u,v,w;
        CLR(num);
        cnt=0;
        for (int i=0;i<n-1;i++) {
            scanf("%d%d%d",&u,&v,&w);
            add_edge(u,v,w);
            add_edge(v,u,w);
        }
        ll ans = 0;
        dfs(1,0);
        for (int i=0;i<cnt;i++) ans += (ll)val[i]*abs(n-2*num[i]);
        printf("%lld",ans/2);
        if (ans) puts("00000000");
        else puts("");
    }
	return 0;
}