Fire POJ - 2152 （树形dp）

题意：给你一颗树，每个点都有一个w值，代表在这个点修建工程需要花费w，还有一个d值，代表离这个点距离小于d的点可以作为这个点的建立工程的点。每条边都有权重，求出覆盖所有点的最小花费。

思路：首先分析，是否具有最优子结构性质，发现任意一颗子树的最小值，都满足最优子结构。考虑动态规划的做法。

这个题的难点在于可以选一个点来覆盖其他点，所以我们的一个想法必然是枚举这个被选取的点。

设dp[u][j]表示以u为根节点的子树，依赖于j节点（其中u节点一定依赖j节点，其他节点可以依赖别的），的最小值。

ans[u]表示覆盖这颗子树的最小值。

这两个状态之间的区别和联系是什么？其实dp[u][j]是为了维护当前子树与它的父亲及以上节点的关系，而ans[u]则是我们的普通动态规划设计的状态。

所以这题的关键在于将dp合理的想ans转化。

dp[u][j]=sum(dp[v][j]-w[j],ans[v])  ,意思是，u点已经确定要选j点了，下面考虑它的子节点是否也选j点就行。

最后右dp[u][j]来更新ans[u]即可。

总结：这题和以前做的树形dp有很大不同，以前做的题目都是一次设计出状态方程就行，但这个因为它同时和子节点，父节点，以及和它相距不超过d的节点都有关系，所以，需要先设出dp[u][j]来解决相距不超过d的问题，再以此来解决ans[u]的问题。

 

代码：

#include <iostream>
#include <stdio.h>
#include <math.h>
#include <string.h>
using namespace std;
const int maxn=1e3+7;
const int inf=0x3f3f3f3f;
int n;
int dp[maxn][maxn],ans[maxn];
int w[maxn],D[maxn];
int head[maxn];
int num;
int d[maxn][maxn];
struct Point
{
    int u,v,w,next;
}edge[maxn<<1];
void addEdge(int u,int v,int w)
{
    edge[num].v=v;
    edge[num].u=u;
    edge[num].w=w;
    edge[num].next=head[u];
    head[u]=num++;
}
void init()
{
    memset(head,-1,sizeof(head));
    memset(d,0,sizeof(d));
    memset(dp,inf,sizeof(dp));
    memset(ans,inf,sizeof(ans));
    num=0;
}

void dfs(int root,int u,int pre)
{
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v, w=edge[i].w;
        if(v==pre) continue;
        d[root][v]=d[v][root]=d[root][u]+w;
        dfs(root,v,u);
    }
}
void solve(int u,int pre)
{
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v;
        if(v==pre) continue;
        solve(v,u);
    }
    for(int i=1;i<=n;i++)
    {
        if(d[u][i]>D[u]) 
        {
            dp[u][i]=inf;
            continue;
        }
        dp[u][i]=w[i];
        for(int j=head[u];j!=-1;j=edge[j].next)
        {
            int v=edge[j].v,W=edge[j].w;
            if(v==pre) continue;
            dp[u][i]+=min(dp[v][i]-w[i],ans[v]);
        }
        ans[u]=min(ans[u],dp[u][i]);
    }
}
int main()
{
    #ifndef ONLINE_JUDGE
        freopen("in.txt","r",stdin);
        freopen("out.txt","w",stdout);
    #endif
    int T;
    cin>>T;
    while(T--)
    {
        init();
        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]);
        }
        for(int i=1;i<n;i++)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            addEdge(u,v,w);
            addEdge(v,u,w);
        }
        for(int i=1;i<=n;i++)
        {
            dfs(i,i,-1);
        }
        solve(1,-1);
        printf("%d\n",ans[1]);
    }
    return 0;
}