洛谷P1441 砝码称重

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。

/*
    看到数据范围那么小，就直接暴力了，但是提高+/省选-的题怎么可能这么好做。。
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 25
using namespace std;
int a[maxn],b[maxn],c[maxn],n,m,tot,ans,ansnow;
bool vis[];
void count(int pos,int sum){
    if(!vis[sum]&&sum!=){
        vis[sum]=;
        ansnow++;
    }
    if(pos==tot+)return;
    count(pos+,sum);
    count(pos+,sum+c[pos]);
}
void dfs(int pos,int cnt){
    if(cnt==m){
        memset(vis,,sizeof(vis));
        ansnow=;
        int num=;
        for(int i=;i<=n;i++)
            if(b[i]!=-)c[++num]=b[i];
        count(,);
        ans=max(ans,ansnow);
        return;
    }
    if(pos==n+)return;
    if(n-pos+<m-cnt)return;
    dfs(pos+,cnt);//不删掉
    b[pos]=-;
    dfs(pos+,cnt+);
    b[pos]=a[pos];
}
int main(){
    scanf("%d%d",&n,&m);tot=n-m;
    for(int i=;i<=n;i++){
        scanf("%d",&a[i]);
        b[i]=a[i];
    }
    dfs(,);
    cout<<ans;
}

60分 暴力TLE

/*
    直接爆搜删去哪些砝码，统计方案数的时候用一个01背包就可以了
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 25
using namespace std;
int a[maxn],b[maxn],c[maxn],n,m,tot,ans,ansnow,f[];
void count(){
    memset(f,,sizeof(f));
    f[]=;
    for(int i=;i<=n-m;i++)
        for(int j=;j>=c[i];j--)
            f[j]+=f[j-c[i]];
    for(int i=;i<=;i++)
        if(f[i])ansnow++;
}
void dfs(int pos,int cnt){
    if(cnt==m){
        ansnow=;
        int num=;
        for(int i=;i<=n;i++)
            if(b[i]!=-)c[++num]=b[i];
        count();
        ans=max(ans,ansnow);
        return;
    }
    if(pos==n+)return;
    if(n-pos+<m-cnt)return;
    dfs(pos+,cnt);//不删掉
    b[pos]=-;
    dfs(pos+,cnt+);
    b[pos]=a[pos];
}
int main(){
    scanf("%d%d",&n,&m);tot=n-m;
    for(int i=;i<=n;i++){
        scanf("%d",&a[i]);
        b[i]=a[i];
    }
    dfs(,);
    cout<<ans;
}

100分 搜索+dp