BZOJ 1060 ZJOI 2007 时态同步 树形DP

题目大意：给出一个有根树，树上的边有边权。现在可以将一个边的边权的值+1，要想使得根节点到所有叶子节点的距离相等，最少需要多少个+1操作。

思路：当只考虑以一个节点为根的子树的时候，就必须要保证这个根节点到所有这个子树中的叶子节点的距离相等，才能保证最终根节点到所有叶子节点的距离都相等。所以就深搜一次，对于每个节点求出到这个节点的最长距离，将所有其他的边加成这个最长距离，记录答案。

但是这样并不能AC，因为标称是错的，它没有开long long ，于是乎我们也要开int才能AC。。-、-

CODE：

#include <cstdio>#include <cstring>#include <iostream>#include <algorithm>#define MAX 500010using namespace std;#define max(a,b) ((a) > (b) ? (a):(b)) int points,root;int head[MAX],total;int next[MAX << 1],aim[MAX << 1],length[MAX << 1]; long long ans; inline void Add(int x,int y,int len){    next[++total] = head[x];    aim[total] = y;    length[total] = len;    head[x] = total;} int DFS(int x,int last){    int max_length = 0;    long long sum = 0,cnt = 0;    for(int i = head[x],l; i; i = next[i]) {        if(aim[i] == last)  continue;        ++cnt;        l = DFS(aim[i],x) + length[i];       （这个l要开成int，不然就是WA）        max_length = max(max_length,l);        sum += l;    }    ans += (max_length * cnt - sum);    return max_length;} int main(){    cin >> points >> root;    for(int x,y,z,i = 1; i < points; ++i) {        scanf("%d%d%d",&x,&y,&z);        Add(x,y,z),Add(y,x,z);    }    DFS(root,0);    printf("%lld\n",ans);    return 0;}