算法设计与分析 3.2 小绵羊

★题目描述

本期活动中,共有N件商品参与促销。顾客可以直接选择M件拿走。

但是,拿走商品有一定的限制,某些商品不能被直接拿走,如果想要拿走他,必须要先拿走他指定的另一件特定商品。

请你计算一下,最多能够拿走总价值为多少的商品。

★输入格式

第一行包括两个正整数N, M,表示共有N件商品,你可以拿走M件。

接下来的N行,每i行包括两个自然数Ai,Bi。

若A = 0, 则表示该商品能直接拿走,若A != 0,则代表你需要先拿走第A个商品。

Bi代表该商品的价值。商品编号从1开始。

输入保证商品之间不会出现环。

★输出格式

输出仅包括一个正整数,表示你能拿走的最大总价值。

★样例输入

3 2
0 1
0 2
0 3

★样例输出

5

★提示

对于100%的数据,1<=N,M<=300,0<=Bi<=1e6.

/*
先虚构出0节点,那么这就是一个完整的树。
约束是只有得到父节点,才能得到子节点。
参考依赖背包的做法,使用邻接表保存这棵树
但是本题中没有背包容量约束,只是限定取个数,因此每次都取价值最大的商品
动态规划使用 F[父节点][个数]:在购买了当前父节点情况下,再买m-1个子节点商品所能得到的最大价值
里对于树中的每个节点来说,就是一个分组背包问题。每个子节点是一组物品。 
我们从叶子节点往上到根节点规划,当规划到 0 节点时,F[0][m]即为我们所求 
	
邻接表
	假设有一棵树如下: 
		    0 
		1   2   3
 	  4 5	6  7 8

	用邻接表可以写成
	 H   NE
	 0   3 2 1 -1
	 1   5 4 -1
	 2   6 -1
	 3   8 7 -1
	 4   -1
*/ 
#include<bits/stdc++.h> 
using namespace std;
int N,M; //物品件数,每人限购个数 
int H[301], NE[301]; //用邻接表保存树结构,H保存邻接表的头,NE是有共同起点下的另一条边 
int W[301]; //物品的价值 
int F[301][301]; //前一维存放父节点、后一维存放每人限购件数 


void dfs(int p){
	for(int i=H[p]; i!=-1; i=NE[i]){ //p是父节点商品序号,所有i都是其子节点商品序号 
		dfs(i);
		for(int m=M; m>=1; --m){//因为要留出一个买父节点p的物品,所以只能分配出2~m个给子节点商品 
			for(int m2=0; m2<=m; ++m2){ //接着再分,每个子节点能得到的分配到的个数是m2 
				F[p][m]=max(F[p][m], F[p][m-m2]+F[i][m2]);
			}
		} 
	}
    for(int m=M; m>=1; --m) F[p][m] = F[p][m-1]+W[p]; //这里要加上P节点的价值哦 
} 

int main(){
	cin>>N>>M;
	
	//构建邻接表 
	memset(H, -1, sizeof(H)); 
	int p; 
	for(int i=1; i<=N; ++i){
		scanf("%d%d",&p,&W[i]);
		NE[i]=H[p];
		H[p]=i;	
	}

	W[0]=0;//虚构0节点
	++M; //因为加上虚构节点,所以购买个数要加1 
	memset(F, 0, sizeof(F));
	dfs(0);
	
	cout<<F[0][M]<<endl;
	return 0;
}

posted on 2019-12-17 23:01  yejifeng  阅读(1399)  评论(0编辑  收藏  举报

导航