题目链接: http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3626
题目大意:树中取点。每过一条边有一定cost,且最后要回到起点。给定预算m,问最大价值。
解题思路:
首先要注意这题要回到起点,由于树的特殊结构(每个结点只有一个父亲)也就是说,要回到开头,
开销是2倍。所以首先m/=2。
然后就是树形背包的求解,这题的cost在边,所以for写法变成如下:
for(m....j....0)
for(0....k....j-e.cost)
dp[i][j]=max(dp[i][j],dp[i][j-k-e.cost]+dp[t][k]);
for循环的主要变化是0的出现,也就是说某些点的开销可以是0(在父亲上算过了)。
所以初始化也要变成:for(int i=0;i<=m;i++) dp[root][i]=w[root];
DP方程的主要变化就是dp[i][j-k] -> dp[i][j-k-e.cost],这里之所以要减去e.cost,是为了防止cost的重复计算。
不妨设k=j-e.cost,你就会发现在计算dp[i][0],这也是为什么要推cost=0这个状态。
#include "cstdio" #include "iostream" #include "cstring" using namespace std; #define maxn 300 struct Edge { int to,next,c; }e[maxn*2]; int w[maxn],num[maxn],dp[maxn][maxn],head[maxn],tol; int n,m,k,u,v,c; void addedge(int u,int v,int c) { e[tol].to=v; e[tol].next=head[u]; e[tol].c=c; head[u]=tol++; } void dfs(int root,int pre) { for(int i=0;i<=m;i++) dp[root][i]=w[root]; int i=root; for(int a=head[root];a!=-1;a=e[a].next) { int t=e[a].to; if(t==pre) continue; dfs(t,root); for(int j=m;j>=0;j--) for(int k=0;k<=j-e[a].c;k++) dp[i][j]=max(dp[i][j],dp[i][j-k-e[a].c]+dp[t][k]); } } int main() { //freopen("in.txt","r",stdin); while(~scanf("%d",&n)) { memset(head,-1,sizeof(head)); memset(dp,0,sizeof(dp)); tol=0; for(int i=1;i<=n;i++) scanf("%d",&w[i]); for(int i=1;i<n;i++) { scanf("%d%d%d",&u,&v,&c); addedge(u,v,c); addedge(v,u,c); } scanf("%d%d",&k,&m); m/=2; dfs(k,k); printf("%d\n",dp[k][m]); } }
2875311 | neopenx | ZOJ | 3626 | Accepted | 636 | 0 | C++ (g++ 4.4.5) | 1174 |
2014-10-22 09:13:13
|
本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系我们删除。