1. 程式人生 > >P1441 砝碼稱重

P1441 砝碼稱重

ron 思路 sca 去掉 class sample ret eof 不同的

題目描述

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

很水的一題,思路可以,dfs+dp,貌似dp可以狀壓……

AC代碼如下:

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int a[22],n,m,ans;
bool  us[2000],f[2000];
void dp()
{
    int res=0,ret=0;
    memset(f,0,sizeof(f));
    f[0]=1;
    for(int i=1;i<=n;i++)
    {
        if(us[i]) continue;
        res+=a[i];
        for(int j=res;j>=0;j--) 
        if(f[j]&&f[j+a[i]]==0) f[j+a[i]]=1,ret++;
    }
    
    ans=max(ans,ret);
    return;
}

void dfs(int id,int used)
{
    if(used>m)return;
    if(id==n+1) {if(used==m) dp();return;}
    us[id]=1;
    dfs(id+1,used+1);
    us[id]=0;
    dfs(id+1,used);
    return;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    scanf("%d",&a[i]);
    dfs(1,0);
    printf("%d",ans);
}

P1441 砝碼稱重