bzoj1060: [ZJOI2007]時態同步(樹形Dp)

Hanks_o發表於2018-03-18

題目傳送門

解法:
很顯然是棵樹。
那麼給定根以後所有的終止節點其實就是葉子節點。

對於任意一個節點。
在其子樹中所有的葉子節點到他的距離應該是一樣的。
因為所有葉子節點走到他之後後面的路程是一樣的。

所以樹形Dp啊。
處理出每個節點下面的葉子節點到他的最長距離。
然後其他的節點往最長距離去補就完了吧。。。

程式碼實現:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
struct node {int x,y,next;ll c;}a[1110000];int len,last[510000],root;
void ins(int x,int y,ll c) {len++;a[len].x=x;a[len].y=y;a[len].c=c;a[len].next=last[x];last[x]=len;}
ll sum[510000],f[510000];  //sum表示這個節點下面的葉子節點到他的最長距離。f表示所有節點弄到路程相等所需要最少的操作次數。
void dfs(int x,int fa) {
    sum[x]=f[x]=0;
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(y!=fa) {dfs(y,x);sum[x]=max(sum[x],sum[y]+a[k].c);}
    }
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(y!=fa) f[x]+=f[y]+sum[x]-(sum[y]+a[k].c);
    }
}
int main() {
    int n;scanf("%d",&n);scanf("%d",&root);
    len=0;memset(last,0,sizeof(last));
    for(int i=1;i<n;i++) {
        int x,y;ll c;scanf("%d%d%lld",&x,&y,&c);
        ins(x,y,c);ins(y,x,c);
    }dfs(root,0);
    printf("%lld\n",f[root]);
    return 0;
}

相關文章