洛谷 P1441 砝码称重

题目描述

现有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。

看到数据不大，想到搜索，搜索去掉的砝码，计算可以称量的重量就是一个01背包。

 #include<iostream>
 #include<cstring>
 #include<cstdio>
 using namespace std;
 int n,m,ans,s,a[],b[],f[];
 void dp()
 {
     memset(f,,sizeof(f));
     f[]=;
     for(int i=;i<=n;i++)
         if(!b[i])
             for(int j=s;j>=a[i];j--)
                 f[j]|=f[j-a[i]];
     int sum=;
     for(int i=;i<=s;i++)
         if(f[i])
             sum++;
     ans=max(ans,sum);
 }
 void dfs(int s,int num)
 {
     if(num>m)
     {
         dp();
         return;
     }
     for(int i=s;i<=n;i++)
     {
         b[i]=;
         dfs(i+,num+);
         b[i]=;
     }
 }
 int main()
 {
     scanf("%d%d",&n,&m);
     for(int i=;i<=n;i++)
     {
         scanf("%d",a+i);
         s+=a[i];
     }
     dfs(,);
     printf("%d\n",ans);
     return ;
 }