洛谷P1441 砝码称重【状态压缩】

时空限制 1000ms / 128MB

题目描述

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

输入格式：

输入文件weight.in的第1行为有两个整数n和m，用空格分隔

第2行有n个正整数a1，a2，a3，……，an，表示每个砝码的重量。

输出格式：

输出文件weight.out仅包括1个整数，为最多能称量出的重量数量。

说明

对于20%的数据，m=0；
对于50%的数据，m≤1；
对于50%的数据，n≤10；
对于100%的数据，n≤20，m≤4，m＜n，ai≤100。

---

题目分析

第一次尝试了STL里面一个很神奇的东西–bitset

#include
#include
#include
#include
#include
#include
#include
using namespace std;
typedef long long lt;

int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}

const int maxn=1100010;
int n,m,ans,a[20];

int main()
{
    n=read();m=read();
    for(int i=1;i<=n;++i) a[i]=read();
    
    for(int i=1;i<=(1<<n)-1;++i)
    {
    	int cnt=0;
        for(int j=1;j<=20;++j)
    	if((1<<j-1)&i) cnt++;
    	if(cnt!=n-m) continue;
        
        bitset<2010> b; b[0]=1;//b某位为1表示该重量可以称出来
        for(int j=1;j<=n;++j)
        if(i&(1<<j-1))
        b|=b<<a[j];//b|=b<
        
        ans=max(ans,(int)b.count()-1);//记得减去重量为0的情况
    }
    printf("%d",ans);
    return 0;
}