首页 > 代码库 > poj2152(Fire) 树形DP
poj2152(Fire) 树形DP
题目链接:http://poj.org/problem?id=2152
题意:一棵带边权的树,边权表示节点间距离,在i上建立消防站的代价是w[i],如果在一点i没建消防站,那么它与距离这个点最近的消防站之间的距离不能大于d[i]。问满足建站最小的花费;
解法;看了陈启峰的论文才会的,感觉挺难的,不过论文里分情况讨论了,应该不需要;dp[i][j]表示在i处选择j处作为供应站(但是并不一定要求是最近的,这样即使有更近的也无所谓,和论文里有出入。),best[i]表示i及其子树满足要求的最小花费,那么转移方程:
best[i]=min(dp[i][j]),1<=j<=n
dp[i][j]=w[j]+min(dp[k][j]-w[j],best[k]),k是i的儿子
代码:
/****************************************************** * @author:xiefubao *******************************************************/ #pragma comment(linker, "/STACK:102400000,102400000") #include <iostream> #include <cstring> #include <cstdlib> #include <cstdio> #include <queue> #include <vector> #include <algorithm> #include <cmath> #include <map> #include <set> #include <stack> #include <string.h> //freopen ("in.txt" , "r" , stdin); using namespace std; #define eps 1e-8 #define zero(_) (abs(_)<=eps) const double pi=acos(-1.0); typedef long long LL; const int Max=100010; const int INF=1e9+7; int n; int cost[1010]; int d[1010]; int dist[1010]; int dp[1010][1010]; int best[1010]; vector<pair<int,int> > vec[1010]; void add(int a,int b,int c) { vec[a].push_back(pair<int,int>(b,c)); vec[b].push_back(pair<int,int>(a,c)); } void getdist(int u,int before,int add) { dist[u]=dist[before]+add; for(int i=0; i<vec[u].size(); i++) if(vec[u][i].first!=before) getdist(vec[u][i].first,u,vec[u][i].second); } void dfs(int u,int before) { //cout<<u<<" "<<before<<endl; for(int i=0; i<vec[u].size(); i++) { if(vec[u][i].first==before) continue; dfs(vec[u][i].first,u); } dist[u]=0; getdist(u,0,0); for(int i=1; i<=n; i++) if(dist[i]<=d[u]) { dp[u][i]=cost[i]; for(int k=0; k<vec[u].size(); k++) if(vec[u][k].first!=before) dp[u][i]+=min(dp[vec[u][k].first][i]-cost[i],best[vec[u][k].first]); best[u]=min(best[u],dp[u][i]); } else dp[u][i]=INF; } int main() { int t; cin>>t; while(t--) { scanf("%d",&n); for(int i=0; i<=n; i++) vec[i].clear(); for(int i=1; i<=n; i++) scanf("%d",cost+i); for(int i=1; i<=n; i++) scanf("%d",d+i); for(int i=0; i<n-1; i++) { int a,b,c; scanf("%d%d%d",&a,&b,&c); add(a,b,c); } for(int i=1; i<=n; i++) best[i]=INF; dfs(1,0); cout<<best[1]<<endl; } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。