01背包查找路径

xiaoxiao2021-02-28  120

做01背包,我们能知道最后的结果,若果求路径,我们可以根据状态转移方程逆推。

看一个百度之星2017年的题

度度熊的午饭时光

Problem 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开始,以空格分隔。

这是一个典型的01背包,需要输出路径。

开一个二维数组vis[][],来标记从是否存在从i->j 的状态,在模拟for循环一遍,记录哪些物品被使用过:

for(int i=n;i>=1;--i) { if(vis[i][s]) { pre[i]=true; s-=c[i]; ++k; } } 最后for循环一遍,输出状态为true的物品就行了。

#include <iostream> #include <stdio.h> #include <string.h> using namespace std; const int maxn=105,maxm=1005; int c[maxn],s[maxn],f[maxm]; bool vis[maxn][maxm],pre[maxn]; int main() { int t,n; scanf("%d",&t); int b; for(int cas=1; cas<=t; ++cas) { scanf("%d%d",&b,&n); for(int i=1; i<=n; ++i) scanf("%d%d",&s[i],&c[i]); memset(f,0,sizeof(f)); memset(pre,false,sizeof(pre)); memset(vis,false,sizeof(vis)); for(int i=1; i<=n; ++i) for(int j=b; j>=c[i]; --j) { if(f[j]<f[j-c[i]]+s[i]) { f[j]=f[j-c[i]]+s[i]; vis[i][j]=true; } else vis[i][j]=false; } int s=b; int k=0; for(int i=n; i>=1; --i) { if(vis[i][s]) { pre[i]=true; s-=c[i]; ++k; } } int sum=0; for(int i=1; i<=n; ++i) if(pre[i]) sum+=c[i]; printf("Case #%d:\n",cas); printf("%d %d\n",f[b],sum); for(int i=1; i<=n; ++i) if(pre[i]) { printf("%d",i); --k; if(!k) printf("\n"); else printf(" "); } } return 0; }

转载请注明原文地址: https://www.6miu.com/read-67321.html

最新回复(0)