描述
小明今天很开心,家里购置的新房就要领钥匙了,新房里有一间他自己专用的很宽敞的房间。更让他高兴的是,妈妈昨天对他说:“你的房间需要购买哪些物品,怎么布置,你说了算,只要不超过N 元钱就行”。今天一早小明就开始做预算,但是他想买的东西太多了,肯定会超过妈妈限定的N 元。于是,他把每件物品规定了一个重要度,分为5 等:用整数1~5 表示,第5 等最重要。他还从因特网上查到了每件物品的价格(都是整数元)。他希望在不超过N 元(可以等于N 元)的前提下,使每件物品的价格与重要度的乘积的总和最大。设第j 件物品的价格为v[j],重要度为w[j],共选中了k 件物品,编号依次为j1...jk,则所求的总和为:v[j1]*w[j1]+..+v[jk]*w[jk]请你帮助金明设计一个满足要求的购物单.
-
输入
-
第一行输入一个整数N(0<N<=101)表示测试数据组数
每组测试数据输入的第1 行,为两个正整数,用一个空格隔开:
N m
(其中N(<30000)表示总钱数,m(<25)为希望购买物品的个数。)
从第2 行到第m+1 行,第j 行给出了编号为j-1
的物品的基本数据,每行有2 个非负整数
v p
(其中v 表示该物品的价格(v≤10000),p 表示该物品的重要度(1~5))
输出
-
每组测试数据输出只有一个正整数,为不超过总钱数的物品的价格与重要度乘积的总和的
最大值(<100000000)
样例输入
-
1 1000 5 800 2 400 5 300 5 400 3 200 2
样例输出
-
3900
最经典的0-1背包问题,这里用dp[i][j]表示在价格不超过j的情况下,前i个物品所能到达的价格与重要度乘积的最大值。
若将第i个物品放入背包,记其价格为money,重要度为value:
dp[i][j]为dp[i-1][j]与dp[i-1][j-money]+value中的较大值,意思是遍历到第i个物品时,如果钱足够,是否将其放入背包,前者为不放,后者为放入。这就是核心操作了orz。
当然前提是j-money大于等于0,代码中需要注意。
#include <cstdio>
int Max(int x,int y)
{
return x>y?x:y;
}
struct M
{
int money;
int value;
}list[30];
int dp[30][30001];
int main()
{
int n;
scanf("%d",&n);
while(n--)
{
int m,N;
scanf("%d%d",&m,&N);
for(int i=1;i<=N;i++)
{
scanf("%d%d",&list[i].money,&list[i].value);
}
for(int i=0;i<=m;i++)
{
dp[0][i]=0;
}
for(int i=1;i<=N;i++)
{
for(int j=m;j>=list[i].money;j--)
{
dp[i][j]=Max(dp[i-1][j],dp[i-1][j-list[i].money]+list[i].value*list[i].money);
}
for(int j=list[i].money-1;j>=0;j--)
{
dp[i][j]=dp[i-1][j];
}
}
printf("%d\n",dp[N][m]);
}
return 0;
}
时间复杂度为O(m*N);分析
dp[i][j]=Max(dp[i-1][j],dp[i-1][j-list[i].money]+list[i].value*list[i].money);
可以发现,dp[i][j]仅与其上一行有关,所以可以压缩为一维数组。
#include <cstdio>
struct M
{
int value;
int money;
}list[26];
int Max(int x,int y)
{
return x>y?x:y;
}
int dp[30001];
int main()
{
int n;
scanf("%d",&n);
while(n--)
{
int N,m;
scanf("%d%d",&N,&m);
for(int i=1;i<=m;i++)
{
scanf("%d%d",&list[i].money,&list[i].value);
}
for(int j=0;j<=N;j++)
{
dp[j]=0;
}
for(int i=1;i<=m;i++)
{
for(int j=N;j>=list[i].money;j--)
{
dp[j]=Max(dp[j],dp[j-list[i].money]+list[i].value*list[i].money);
}
}
printf("%d\n",dp[N]);
}
return 0;
}
时间复杂度为O(N)。
ps:dp数组应设置成全局变量,否则不够用。
runTimeError多是因为数组开的略小造成的。