记忆背包

题目:https://vijos.org/p/1071

描述

过年的时候,大人们最喜欢的活动,就是打牌了。xiaomengxian不会打牌,只好坐在一边看着。

这天,正当一群人打牌打得起劲的时候,突然有人喊道:“这副牌少了几张!”众人一数,果然是少了。于是这副牌的主人得意地说:“这是一幅特制的牌,我知道整副牌每一张的重量。只要我们称一下剩下的牌的总重量,就能知道少了哪些牌了。”大家都觉得这个办法不错,于是称出剩下的牌的总重量,开始计算少了哪些牌。由于数据量比较大,过了不久,大家都算得头晕了。

这时,xiaomengxian大声说:“你们看我的吧!”于是他拿出笔记本电脑,编出了一个程序,很快就把缺少的牌找了出来。

如果是你遇到了这样的情况呢?你能办到同样的事情吗?

格式

输入格式

第一行一个整数TotalW,表示剩下的牌的总重量。

第二行一个整数N(1<N<=100),表示这副牌有多少张。

接下来N行,每行一个整数Wi(1<=Wi<=1000),表示每一张牌的重量。

输出格式

如果无解,则输出“0”;如果有多解,则输出“-1”;否则,按照升序输出丢失的牌的编号,相邻两个数之间用一个空格隔开。

样例1

样例输入1

270
4
100
110
170
200

样例输出1

2 4

限制

各个测试点1s

提示

Sample input #2
270
4
100
110
160
170

Sample output #2
-1

Sample input #3
270
4
100
120
160
180

Sample output #3
0

题解:背包,n张牌,m最大重量,传统思路做这道题,符合重量的就加入,但是要按它的要求输出,注意两个点,一个是无解多解,一个是唯一解,一张一张牌放到称上,1-n依次张牌的重量为a[i],一个fa[i]数组用来记录为i的重量是由哪些牌填充的(代码可以看出),dp[i-a[i]](上一张牌的转移),如果上一张牌已经由当前其他的牌的重量填充过了,也就是已经可以组成最大的m重量了,就标记dp[i]为-1,dp下去就可以了,还有一个特判就是m==0的时候注意一下就可以了!

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<string>
#include<vector>
#include<queue>
#include<set>
#include<map>
#include<cmath>
#define mst(ss,b) memset((ss),(b),sizeof(ss))
#define maxn 0x3f3f3f3f
using namespace std;
int n,m;
int dp[100001];
int vis[100001];
int fa[100001];
int a[110];
int main()
{
    cin.sync_with_stdio(false);
    cin>>m>>n;
    mst(a,0);
    mst(vis,0);
    mst(fa,0);
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    dp[0]=1;
    if(m==0)
    {
        for(int i=1;i<n;i++)
            cout<<i<<" ";
        cout<<n<<endl;
        return 0;
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=m;j>=a[i];j--)
        {
            if(dp[j-a[i]])
            {
                if(dp[j]==0)
                {
                    dp[j]=1;
                    fa[j]=i;
                }
                else
                    dp[j]=-1;
            }
            if(dp[j-a[i]]==-1)
            {
                dp[j]=-1;
            }
        }
    }
    //cout<<dp[m]<<endl;
    if(dp[m]==0 || dp[m]==-1)
    {
        cout<<dp[m]<<endl;
        return 0;
    }
    int temp=m;
    while(temp)
    {
        vis[fa[temp]]=1;
        temp=temp-a[fa[temp]];
    }
    for(int i=1;i<=n;i++)
    {
        if(!vis[i])
            cout<<i<<" ";
    }
    return 0;
}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值