给出扑克丢失后的总重量,给出几张扑克的编号和它们的重量,让你指出丢失了哪几张扑克。如果无解输出0,多种丢失方式输出-1,只有一种丢失方式则输出丢失的牌的编号。
方法:dp[i]表示重量为i的组合方式总数,如果dp[v] == 0 则无解。如果等于1,则只有一种组合方式,否则,多种解。dp[v] == 1的时候需要输出编号,开一个前驱数组记录一下。
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include<set>
using namespace std;
int dp[100002],path[100002];//前驱数组
int a[102],ans[102];
set<int> s;
int main()
{
int V,n;
scanf("%d%d",&V,&n);
for(int i = 1 ; i <= n ; i++)
scanf("%d",&a[i]);
memset(dp,0,sizeof(dp));
dp[0] = 1;
for(int i = 1 ; i <= n ; i++)
{
for(int j = V ; j >= a[i] ;j--)
{
dp[j] += dp[j-a[i]];
if(dp[j-a[i]]&&!path[j])
path[j] = i;//记录前驱
}
}
if(dp[V] == 0) printf("0\n");
else if(dp[V] == 1)
{
for(int i=n;i>=1;i--)
{
if(path[V]==i)
{
s.insert(a[i]);//找到前驱,将前驱装到set里。
V=V-a[i];
}
}
int k = 0;
for(int e = 1 ; e <=n ; e++)
if(s.count(a[e])==0)
ans[k++] = e;
for(int e = 0 ; e < k-1 ; e++)
printf("%d ",ans[e]);
printf("%d\n",ans[k-1]);
}
else printf("-1\n");
return 0;
}