新年趣事之打牌 记录路径+数量判断的01背包问题

描述

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

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

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

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

格式

输入格式

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

第二行一个整数N（1<N<=100），表示这副牌有多少张。

接下来N行，每行一个整数Wi（1<=Wi<=1000），表示每一张牌的重量。

输出格式

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

样例1

样例输入1

270
4
100
110
170
200

 
 

  
  

   
   Copy
  
  
 
 

样例输出1

2 4

 
 

  
  

   
   Copy
  
  
 
 

限制

各个测试点1s

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
#include<set>
#include <assert.h>
#include<iostream>
using namespace std;
int a[111];
struct node
{
    int id, pre;//id表示当前状态我们是加了id后凑出来的。 pre表示上一个是啥
    int chuxiancishu;
}dp[110000];
set<int>ans;
int main()
{
    int val,n;
    cin >>val>>n;
    int sum = 0;
    for (int i = 1; i <= n; ++i)
    {
        cin >> a[i];
        sum += a[i];
    }
    dp[0].chuxiancishu = 1;
    dp[0].id = dp[0].pre = 1e9;
    val=sum-val;//我们把题转换成从n个物体凑出val
    for (int i = 1; i <= n;i++)
    {
        for (int j = val; j >= a[i];j--)
        {
            if(dp[j-a[i]].chuxiancishu>0)//之前的出现过
            {
                 if(dp[j].chuxiancishu==0)//当前没有出现过
                 {
                     dp[j].chuxiancishu=1;
                     dp[j].id=i;
                     dp[j].pre=j-a[i];
                 }
                 else if(dp[j].chuxiancishu)//这个已经存在了
                 {
                      dp[j].chuxiancishu =2;
                 }
            }
        }
    }
    if (dp[val].chuxiancishu == 0)
    {
        cout << 0 << endl;
        return 0;
    }
    if (dp[val].chuxiancishu == 2)
    {
        cout << -1 << endl;
        return 0;
    }
    int t = val;
    while (t != 0)
     {
        if (dp[t].id == 1e9)
             break;
        ans.insert(dp[t].id);
        t = dp[t].pre;
    }
    for (set<int> :: iterator it = ans.begin(); it != ans.end(); ++it) {
        cout << *it << " ";
    }
}