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
#include

using namespace std;

#define max_number 100
#define max_people 1000

int dp[max_number][max_people]; //dp[i][j]表示使用 j 个人开采1~i座金矿的最大收益

int DP(int gold[],int peopleNeed[],int n,int people)
{
for(int i=n;i>=1;i–) //从第n座到第1座金矿，逐步推导出答案
{
for(int j=0;j<=people;j++) //j代表使用的人数
{
if(j>=peopleNeed[i]) //当前人数够开第i座矿
{
int dig=dp[i+1][j-peopleNeed[i]]+gold[i]; //对于第i座金矿，比较挖与不挖两种情况下的收益
int nodig=dp[i+1][j];
dp[i][j]=max(dig,nodig); //调用max函数选取收益大的一个
}
else // 人数不够，放弃开采
{
dp[i][j]=dp[i+1][j];
}
}
}
}
int main()
{
int n,people;
int gold[max_number],peopleNeed[max_number]; //gold[]为每座金矿黄金储量，peopleNeed[]为挖每座金矿所需人数
gold[0]=0;
cout<<“输入金矿数” <<endl;
cin>>n;
cout<<“输入每座金矿的黄金储量 " <<endl; //输入金矿数以及每座金矿的黄金储量
for(int i=1;i<=n;i++) //从下标1开始输入
{
cin>>gold[i];
}
cout<<“输入总人数”<<endl;
cin>>people;
cout<<“输入每座金矿所需的人数 “<<endl; //输入总人数以及挖每座金矿所需的人数
for(int i=1;i<=n;i++) //从下标1开始遍历
{
cin>>peopleNeed[i];
}
DP(gold,peopleNeed,n,people);
cout<<endl;
cout<<“输出表格如下：”<<endl;
cout<<”-------------------------------------”<<endl;
for(int i=1;i<=n;i++)
{
for(int j=0;j<=people;j++)
{
cout<<dp[i][j]<<” “;
}
cout<<endl;
cout<<endl;
}
cout<<”-------------------------------------"<<endl;
cout<<“最多可获得黄金数量为：”;
cout<<dp[1][people]<<endl;

return 0;

}
运行结果：