题意:
给一颗树可以在树上的节点上建立消防站费用为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;
}