Description
度度熊最期待每天的午饭时光,因为早饭菜品清淡,晚饭减肥不敢吃太多(胖纸的忧伤T.T)。
百度食堂的午餐超级丰富,祖国各大菜系应有尽有,度度熊在每个窗口都有爱吃的菜品,而且他还为喜爱的菜品打了分,吃货的情怀呀(>.<)。
但是,好吃的饭菜总是很贵,每天的午饭预算有限,请帮度度熊算一算,怎样打饭才能买到的最好吃的饭菜?(不超过预算、不重样、午餐等分最高的情况下,选择菜品序号加和最小,加和相等时字典序最小的组合)
Input
第一行一个整数T,表示T组数据。
每组测试数据将以如下格式从标准输入读入:
B
N
score_1 cost_1
score_2 cost_2
:
score_N cost_N
第一行,正整数B(0 <= B <= 1000),代表午餐的预算。
第二行,正整数N (0 <= N <= 100),代表午餐可选的菜品数量
从第三行到第 (N + 2) 行,每行两个正整数,以空格分隔,score_i表示菜品的得分,cost_i表示菜品的价格(0 <= score_i, cost_i <= 100)。
Output
对于每组数据,输出两行:
第一行输出:”Case #i:”。i代表第i组测试数据。
第二行输出菜品的总得分和总花费,以空格分隔。
第三行输出所选菜品的序号,菜品序号从1开始,以空格分隔。
Sample Input
2
29
6
9 10
3 4
6 5
7 20
10 9
15 11
0
2
2 23
10 12
Sample Output
Case #1:
34 29
2 3 5 6
Case #2:
0 0
Solution
01背包问题,要求所选物品编号和最小,编号和相同时选字典序小的
把物品编号反过来,
dp[i][j]
表示前
i
个物品花费为
时间复杂度O(NB)
Code
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int INF=0x3f3f3f3f,maxn=101,maxb=1001;
int T,n,b,s[maxn],c[maxn],dp[maxn][maxb],sum[maxn][maxb];
int main()
{
int Case=1;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&b,&n);
for(int i=n;i>=1;i--)scanf("%d%d",&s[i],&c[i]);
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)
for(int j=1;j<=b;j++)
sum[i][j]=INF;
for(int i=0;i<=n;i++)sum[i][0]=0;
for(int i=0;i<=b;i++)sum[0][i]=0;
for(int i=1;i<=n;i++)
for(int j=b;j>=0;j--)
{
dp[i][j]=dp[i-1][j],sum[i][j]=sum[i-1][j];
if(j>=c[i])
{
if(dp[i][j]<dp[i-1][j-c[i]]+s[i])
dp[i][j]=dp[i-1][j-c[i]]+s[i],sum[i][j]=sum[i-1][j-c[i]]+(n+1-i);
else if(dp[i][j]==dp[i-1][j-c[i]]+s[i])
sum[i][j]=min(sum[i][j],sum[i-1][j-c[i]]+(n+1-i));
}
}
printf("Case #%d:\n",Case++);
if(dp[n][b]==0)printf("0 0\n");
else
{
int cost=0,bb=b;
vector<int>ans;
for(int i=n;i>=1;i--)
if(dp[i][bb]==dp[i-1][bb-c[i]]+s[i]&&sum[i][bb]==sum[i-1][bb-c[i]]+(n+1-i))
ans.push_back(n+1-i),cost+=c[i],bb-=c[i];
printf("%d %d\n",dp[n][b],cost);
for(int i=0;i<ans.size();i++)printf("%d%c",ans[i],i==ans.size()-1?'\n':' ');
}
}
return 0;
}