0-1背包问题 题目:国王和金矿问题 描述:有一个国家发现了max_n座金矿,参与挖矿工人的总数是max_people人。每座金矿的黄金储量不同为一维数组gold[],需要参与挖掘的工人数也不同为一维

题目四:0-1背包问题

题目:国王和金矿问题

描述:有一个国家发现了max_n座金矿,参与挖矿工人的总数是max_people人。每座金矿的黄金储量不同为一维数组gold[],需要参与挖掘的工人数也不同为一维数组peopleNeed[]。每座金矿要么全挖,要么不挖,不能派出一半人挖取一半金矿。要想得到尽可能多的黄金,应该选择挖取哪几座金矿?
功能:
  (1) 要求max_n、max_people、gold和ppeopleNeed均为可输入的;
  (2) 编写DP函数,求解答案F;
  (3) 编写main主函数,完成输入,调用DP函数和显示答案结果。

样例输入1:
5
92 22 87 46 90
100
77 22 29 50 99

样例输出1:
133


代码:

#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
int main() {
	
	int max_n;   cin>>max_n;	//输入金矿 
	int *gold = new int[max_n+1];	//金矿数组,从1开始 
	int *peopleNeed = new int[max_n+1];	//每个矿需要的人,从1开始 
		
	for(int i = 1 ; i <= max_n ; i ++) cin>>gold[i];	//每个金矿容量 
		
	int max_people;
	cin>>max_people;		//输入人的总数 
	
	int **dp = new int *[max_n+1];		//先用指针定义,横坐标是人, 纵坐标是矿 
	for(int i = 0; i < max_n+1; i++) 	//开辟空间 
		dp[i] = new int[max_people+1];
	
	
	for(int i = 0 ; i <= max_people ; i++) 	//第一列均为0 
		dp[0][i] = 0;
	
	
	for(int i = 1 ; i <= max_n ; i ++) 	 
		cin>>peopleNeed[i];
	
	
	for(int i = 1 ; i <= max_n ; i++) 
		for(int j = 0 ; j <= max_people ; j++) 
			if(peopleNeed[i]<=j) 	//当某矿需要人数<=现有人数时,
			//最大值等于 没加这座矿以前这些人能开的最大值,和加上这座矿以后减去对应的人数的最大值比较 
				dp[i][j] = max(dp[i-1][j],dp[i-1][j-peopleNeed[i]]+gold[i]);	
			else 	//如果>= 说明该矿开不了, 还等于原来的。 
				dp[i][j] = dp[i-1][j];
	
	cout<<dp[max_n][max_people]<<endl;		//输出最后一行最后一列即可 
	
	
	
	return 0;
 } 

时间复杂度: O(mn),m为金矿数量,n为人数。

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 像素格子 设计师:CSDN官方博客 返回首页