原本以为这题就是简单一搜索,直接套DFS模板发现样例也能过,但后来发现只是凑巧正确。因为题目要求的是选出n-m个后,再在n-m个当中选若干个并计算能出现多少个和。
选哪n-m个不确定,这部分不得不爆搜(至少我只会这样)。那在n-m个中选的过程难道也只能爆搜了吗?
显然不是,我们可以用DP完成这一过程(01背包)。
代码如下:
#include <cstdio>
#include <algorithm>
const int M=22,MAXW=2001;
using namespace std;
int a[M],n,sum;
int vis[M]; //已选的数的集合
inline int dpCount(int tot) {
static bool dp[MAXW];
int ret=0;
for(int i=1; i<=sum; i++) dp[i]=0;
dp[0]=1;
for(int i=1; i<=tot; i++) {
for(int j=sum; j>=vis[i]; j--) {
dp[j]|=dp[j-vis[i]];
}
}
for(int i=1; i<=sum; i++) {
if(dp[i]) ret++;
}
return ret;
}
int dfs(int tot,int st=0,int have=0) {
if(have==tot) {
return dpCount(tot);
}
int res=0;
for(int i=st; i<n; i++) {
vis[have+1]=a[i];
res=max(res,dfs(tot,i+1,have+1));
}
return res;
}
int main() {
int m;
scanf("%d%d",&n,&m);
for(int i=0; i<n; i++){
scanf("%d",&a[i]);
sum+=a[i];
}
printf("%d",dfs(n-m));
return 0;
}