题目描述
现有n个砝码,重量分别为a1,a2,a3,……,an,在去掉m个砝码后,问最多能称量出多少不同的重量(不包括0)。
请注意,砝码只能放在其中一边。
输入格式
输入文件weight.in的第1行为有两个整数n和m,用空格分隔
第2行有n个正整数a1,a2,a3,……,an,表示每个砝码的重量。
输出格式
输出文件weight.out仅包括1个整数,为最多能称量出的重量数量。
输入输出样例
输入 #1
3 1
1 2 2
输出 #1
3
说明/提示
【样例说明】
在去掉一个重量为2的砝码后,能称量出1,2,3共3种重量。
【数据规模】
对于20%的数据,m=0;
对于50%的数据,m≤1;
对于50%的数据,n≤10;
对于100%的数据,n≤20,m≤4,m<n,ai≤100。
解释:枚举选中的,然后进行dp,dp[i]:重量为i是否存在,最后再统计一下就好了
#include<iostream>
#include<cstring>
using namespace std;
bool dp[2003]={0};
int n=0,m=0,a[23]={0};
int vec[23]={0};
bool H[23]={0};
int t=0;
int ret=0;
void C(int pos){
if(n-pos+1+t<m) return;
if(pos==n+1){
memset(dp,0,sizeof(dp));
memset(H,0,sizeof(H));
for(int i=1;i<=t;i++) H[vec[i]]=1;
dp[0]=1;
int sum=0;
for(int i=1;i<=n;i++){
if(H[i]) continue;
for(int j=2000;j>=0;j--) if(j-a[i]>=0) dp[j]|=dp[j-a[i]];
}
for(int i=1;i<=2000;i++) if(dp[i]) sum++;
ret=max(ret,sum);
return;
}
C(pos+1);
if(t>=m) return;
vec[++t]=pos;
C(pos+1);--t;
}
int main(){
ios::sync_with_stdio(false);
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>a[i];
C(1);
cout<<ret<<endl;
return 0;
}