首页 > 代码库 > BZOJ 1060 ZJOI 2007 时态同步 树形DP

BZOJ 1060 ZJOI 2007 时态同步 树形DP

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


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

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


CODE:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define MAX 500010
using 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;
}


BZOJ 1060 ZJOI 2007 时态同步 树形DP