poj 2152 树形dp（建立消防站）

题意：给定n个节点组成的树，树边有权。现在要在一些点上建立消防站，每个点建站都有花费cost[i]。如果某点上没有消防站，那么在距离它d[i]之内的某个点上必须有消防站。求符合上述条件的最小费用建站方案。n <= 1000。

思路：知道是树形dp，但是该怎么做一直想不出来。只能看别人的题解，发现在陈启峰2006年的《一张一弛，解题之道 ——“约制、放宽”方法在解题中的应用》论文中提到了这道题的解法。大题思路是：设dp[i][j]表示i点及其子树都符合情况下，并且i点依赖j点的最小花费。best[i]表示以i为根的子树符合题目要求的最小花费。这样状态转移方程就是dp[i][j] = cost[j] + sum(min(dp[k][j]-cost[j],best[k])) (k为i的子节点，j为我们枚举的n个点)，因为i的每个子节点可以和i一样依赖j结点（也即best[k]在dp[k][j]处取到），那么花费是dp[k][j]-cost[j]，或者依赖以k为根的树中的某点，花费是best[k]，最后再加上cost[j]，因为要在j结点建站所以要增加花费。

比较费解的是求两点间距离时一开始写的bfs（代码中的bfs2），C++交会TLE，于是改成dfs（代码中的bfs)来写就过了。但是后来试了一下bfs用g++交是能够A的。

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <queue>
using namespace std;
#define INF 0x3fffffff
#define clr(s,t) memset(s,t,sizeof(s))
#define N 1005
int w[N],d[N],dis[N][N],dp[N][N],first[N],res[N];
struct edge{
    int y,c,next;
}e[N<<1];
int T,n,top;
void add(int x,int y,int c){
    e[top].y = y;
    e[top].c = c;
    e[top].next = first[x];
    first[x] = top++;
}
void bfs(int s,int x,int fa){
    for(int i = first[x];i!=-1;i=e[i].next)
        if(e[i].y != fa){
            dis[s][e[i].y] = dis[s][x]+e[i].c;
            bfs(s,e[i].y,x);
        }
}
void bfs2(int x){
     queue<int> q;
     int i,now;
     q.push(x);
     dis[x][x] = 0;
     while(!q.empty()){
         now = q.front();
         q.pop();
         for(i = first[now];i!=-1;i=e[i].next)
             if(dis[x][e[i].y]==-1){
                 dis[x][e[i].y] = dis[x][now]+e[i].c;
                 q.push(e[i].y);
             }
     }
}
void dfs(int x,int fa){
    int i,j,y;
    for(i = first[x];i!=-1;i=e[i].next)
        if(e[i].y != fa)
            dfs(e[i].y,x);
    for(i = 1;i<=n;i++){
        dp[x][i] = INF;
        if(dis[x][i] <= d[x]){
            dp[x][i] = w[i];
            for(j = first[x];j!=-1;j=e[j].next){
                y = e[j].y;
                if(y==fa)
                    continue;
                dp[x][i] += min(dp[y][i]-w[i],res[y]);
            }
        }
        res[x] = min(res[x],dp[x][i]);
    }
}
int main(){
    scanf("%d",&T);
    while(T--){
        int i,a,b,c;
        clr(first, -1);
        top = 0;
        scanf("%d",&n);
        for(i = 1;i<=n;i++)
            res[i] = INF;
        for(i = 1;i<=n;i++)
            scanf("%d",&w[i]);
        for(i = 1;i<=n;i++)
            scanf("%d",&d[i]);
        for(i = 1;i<n;i++){
            scanf("%d %d %d",&a,&b,&c);
            add(a,b,c);add(b,a,c);
        }
        
        /*clr(dis, 0);
        for(i = 1;i<=n;i++)
            bfs(i,i,0);*/
        
        clr(dis, -1);
        for(i = 1;i<=n;i++)
            bfs2(i);
        dfs(1,-1);
        printf("%d\n",res[1]);
    }
    return 0;
}