POJ 2152 树形DP+暴力枚举

题目链接

题意：

给一颗树可以在树上的节点上建立消防站费用为W[i]，如果不建消防站需要在距离该节点距离小于等于D[i]的地方有消防站，求使得整颗树被覆盖的最小费用。

思路：

定义ans数组：ans[u]表示以u为根的子树的答案

定义dp数组：dp[u][v]表示节点u被建立在节点v的消防站覆盖的最小答案

定义dis数组：dis[u][v]表示u，v两点之间的距离

此题数据较小为1000比正常的树形DP题的数据差一个平方，所以在一些处理上可以考虑枚举

对于dp[u][v]的计算：

当dis[u][v]>D[u]时，dp[u][v]=inf 反之

dis[u][v] = W[v] + Σ min( ans[son] ,  dp[son][v]-W[v]{当son可以被v保护时} )

dp[son][v]-W[v]可以看成是否可以更加优化答案

ans[u] = min( dp[u][1] , ... , dp[u][n] )

C++代码：

#include<map>
#include<set>
#include<stack>
#include<cmath>
#include<queue>
#include<vector>
#include<string>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 1010;
const int maxm = 2010;
const int  inf = 0x3f3f3f3f;

int n,tol,head[maxn],W[maxn],D[maxn];
struct edge
{
    int to,cost,next;
}es[maxm];

void addedge( int u , int v , int w )
{
    es[tol].to = v;
    es[tol].cost = w;
    es[tol].next = head[u];
    head[u]  = tol++;
}

int ans[maxn],dp[maxn][maxn],dis[maxn][maxn];

void cal_dis( int s , int u , int f , int d )
{
    dis[s][u] = d;
    for ( int i=head[u] ; i!=-1 ; i=es[i].next )
        if ( es[i].to!=f ) cal_dis( s , es[i].to , u , d+es[i].cost );
}

void dfs( int u , int f )
{
    ans[u] = inf;
    for ( int i=head[u] ; i!=-1 ; i=es[i].next )
        if ( es[i].to!=f ) dfs( es[i].to , u );
    for ( int v=1 ; v<=n ; v++ )
    {
        if ( dis[u][v]>D[u] )
            dp[u][v] = inf;
        else
        {
            dp[u][v] = 0;
            for ( int i=head[u] ; i!=-1 ; i=es[i].next )
            {
                if ( es[i].to!=f )
                {
                    if ( dp[es[i].to][v]!=inf )
                        dp[u][v] += min( dp[es[i].to][v]-W[v] , ans[es[i].to] );
                    else
                        dp[u][v] += ans[es[i].to];
                }
            }
            dp[u][v] += W[v];
        }
    }
    for ( int i=1 ; i<=n ; i++ )
        ans[u] = min( ans[u] , dp[u][i] );
}

int main()
{
    int T; scanf( "%d" , &T );
    while( T-- )
    {
        scanf ( "%d" , &n );
        tol = 0; memset ( head , -1 , sizeof(head) );
        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=2 ; 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++ )
            cal_dis( i , i , 0 , 0 );
        dfs( 1 , 0 );
        printf ( "%d\n" , ans[1] );
    }
    return 0;
}