01背包查找路径

做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;
}



  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值