POJ 2152 Fire(树形DP)

题意：

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

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

好难的题。。。

 #include<algorithm>
 #include<cstdio>
 #include<cmath>
 #include<cstring>
 #include<iostream>
 int tot,go[],first[],next[],val[];
 int dis[],f[][];
 int n,w[],d[],best[];
 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]!=-) 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=;i<=n;i++) dis[i]=-;
     dis[x]=;
     Dfs(x);best[x]=;
     for (int i=;i<=n;i++) f[x][i]=;
     for (int i=;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=;i<=n;i++) scanf("%d",&w[i]);
         for (int i=;i<=n;i++) scanf("%d",&d[i]);
         tot=;
         for (int i=;i<=n;i++) first[i]=;
         for (int i=;i<n;i++){
             int x,y,z;
             scanf("%d%d%d",&x,&y,&z);
             add(x,y,z);
         }
         dfs(,);
         printf("%d\n",best[]);
     }
 }