描述
过年的时候,大人们最喜欢的活动,就是打牌了。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;
}