洛谷 1441 砝码秤重

【题意概述】

 现有n个砝码，重量分别为a1，a2，a3，……，an，在去掉m个砝码后，问最多能称量出多少不同的重量（不包括0）。

【题解】

 先DFS决定用哪些砝码，然后DP求可以称出的重量有多少种

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=2010;
int n,m,a[maxn],b[maxn],ans=0,tot;
bool v[maxn],f[maxn];
void read(int &k){
    k=0; int f=1; char c=getchar();
    while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    k*=f;
}
void dp(){
    memset(f,0,sizeof(f)); tot=1; b[1]=0;
    for (int i=1;i<=n;i++) if(v[i]){
        int now=tot;
        for(int j=1;j<=now;j++) if(!f[b[j]+a[i]]) f[b[j]+a[i]]=1,b[++tot]=b[j]+a[i];
        ans=max(ans,tot-1);
    }
}
void dfs(int dep,int tot){
    if(dep>=n){if(tot>=n-m) dp(); return;}
    if(tot<n-m) v[dep+1]=1,dfs(dep+1,tot+1),v[dep+1]=0;
    dfs(dep+1,tot);
}
int main(){
    read(n); read(m);
    for (int i=1;i<=n;i++) read(a[i]);
    sort(a+1,a+1+n); dfs(0,0);
    return printf("%d
",ans),0;
}