POJ 2152 Fire(树形DP)

题意：

思路：令F[i][j]表示

的最小费用。Best[i]表示以i为根节点的子树多有节点都找到负责消防站的最小费用。

好难的题。。。

#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
int tot,go[200005],first[200005],next[200005],val[200005];
int dis[200005],f[2005][2005];
int n,w[200005],d[200005],best[200005];
void insert(int x,int y,int z){
    tot++;
    go[tot]=y;
    next[tot]=first[x];
    first[x]=tot;
    val[tot]=z;
}
void add(int x,int y,int z){
    insert(x,y,z);
    insert(y,x,z);
}
void Dfs(int x){
    for (int i=first[x];i;i=next[i]){
        int pur=go[i];
        if (dis[pur]!=-1) continue;
        dis[pur]=dis[x]+val[i];
        Dfs(pur);
    }
}
void dfs(int x,int fa){
    for (int i=first[x];i;i=next[i]){
        int pur=go[i];
        if (pur==fa) continue;
        dfs(pur,x);
    }
    for (int i=1;i<=n;i++) dis[i]=-1;
    dis[x]=0;
    Dfs(x);best[x]=99999999;
    for (int i=1;i<=n;i++) f[x][i]=99999999;
    for (int i=1;i<=n;i++)
     if (dis[i]<=d[x]){
            f[x][i]=w[i];
            for (int j=first[x];j;j=next[j]){
                int pur=go[j];
                if (pur==fa) continue;
                f[x][i]+=std::min(best[pur],f[pur][i]-w[i]);
            }
            best[x]=std::min(best[x],f[x][i]);
    }
}
int main(){
    int T;
    scanf("%d",&T);
    while (T--){
        scanf("%d",&n);
        for (int i=1;i<=n;i++) scanf("%d",&w[i]);
        for (int i=1;i<=n;i++) scanf("%d",&d[i]);
        tot=0;
        for (int i=1;i<=n;i++) first[i]=0;
        for (int i=1;i<n;i++){
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            add(x,y,z);
        }
        dfs(1,0);
        printf("%d
",best[1]);
    }
}