POJ 1947 树形 DP + 分组背包

来源:互联网 发布:cepii baci数据库 编辑:程序博客网 时间:2024/05/13 03:11

做这个题的时候实在太虚了。。。

状态都搞不清楚怎么来的。

首先,我们不知道到底我们切出来的那棵树的树根到底是哪个。所以,有必要枚举一下树根。这样足以表示所有状态

然后,对于每个树根来说,他做的是一个 分组背包,每个分组里面只能取一个物品的分组背包。而且,第一个物品很特殊——树根。如果不选这个,则下面的都不能选。既然我们的状态表示以此为根的子树,所以这个点,必须选。

费用中的减一。。因为我们是从没有子树的状态开始的,再加上子树就相当于少用了一刀。

这个分组背包是每组物品只能选择一个,所以对每个价格用组内所有的物品更新一遍。

对于一组内可以选择多个的,要使用本组内每个物品对价格更新一遍

#include <stdio.h>#include <iostream>#include <queue>#include <algorithm>#include <map>#include <vector>#include <cmath>#include <string.h>#include <stdlib.h>#include <time.h>#include <fstream>#include <set>#include <stack>using namespace std;#define READ freopen("acm.in","r",stdin)#define WRITE freopen("acm.out","w",stdout)#define ll long long#define ull unsigned long long #define PII pair<int,int>#define PDI pair<double,int>#define PDD pair<double,double>#define MII map<int,int>::iterator #define fst first#define sec second#define MS(x,d) memset(x,d,sizeof(x))#define INF 0x3f3f3f3f#define ALL(x) x.begin(),x.end()#define lson l,m,rt<<1#define rson m+1,r,rt<<1|1#define ROOT 0,n-1,1#define PB push_back#define FOR(a,b,c) for(int a=b;a<c;a++)#define MOD 1000000007#define keyTree (ch[ ch[root][1] ][0])#define MAX 1111#define SIG 128int n,p;int dp[155][155];vector<int> G[155];int dfs(int u,int fa){int sum=0;for(int i=0;i<G[u].size();i++){int v=G[u][i];if(v!=fa){sum++;dfs(v,u);}}dp[u][1]=sum;for(int i=0;i<G[u].size();i++){int v=G[u][i];if(v!=fa)for(int j=p;j>=2;j--)// fa u {for(int k=1;k<j;k++)// child vif(dp[u][j-k]!=INF&&dp[v][k]!=INF)dp[u][j]=min(dp[u][j],dp[u][j-k]+dp[v][k]-1);}}return 0;}int main(){READ;while(scanf("%d%d",&n,&p)!=EOF){MS(dp,0x3f);for(int i=0;i<155;i++)G[i].clear();for(int i=1;i<n;i++){int f,t;scanf("%d%d",&f,&t);G[f].PB(t),G[t].PB(f);}dfs(1,-1);int ans=INF;for(int i=1;i<=n;i++)ans=min(ans,dp[i][p]+!(i==1));cout<<ans<<endl;}return 0;}


0 0
原创粉丝点击