Relocation - POJ 2923（状态压缩+01背包）

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
using namespace std;

const int MAXN = 1<<10;
const int Bit = 10;
const int oo = 1e9+7;

int N, Ca, Cb;

bool Judge(int x, int w[])
{///判断状态为x的时候是否能一次运输完毕
    int v[107]={1}, sum=0;

    for(int i=0; i<N; i++)if(x & (1<<i))
    {
        sum += w[i];
        for(int j=Ca-w[i]; j>=0; j--)
        {
            if(v[j])v[j+w[i]] = true;
        }
    }
    if(sum > Ca+Cb)return false;

    for(int i=0; i<=Ca; i++)
    {
        if(v[i] && Cb >= sum-i)
            return true;
    }

    return false;
}

int main()
{
    int T, t=1;

    scanf("%d", &T);

    while(T--)
    {
        int i, j, M, w[MAXN], f[MAXN], k=0;

        scanf("%d%d%d", &N, &Ca, &Cb);

        for(i=0; i<N; i++)
            scanf("%d", &w[i]);

        for(i=0, M=1<<N; i<M; i++)
        {
            if(Judge(i, w))f[k++] = i;
        }

        int dp[MAXN]={0};

        for(i=1; i<M; i++)
            dp[i] = oo;

        for(i=0; i<k; i++)
        for(j=0; j<M; j++)
        {
            if((j&f[i]) == 0)
            {
                dp[j|f[i]] = min(dp[j|f[i]], dp[j]+1);
            }
        }

        printf("Scenario #%d:\n%d\n\n", t++, dp[M-1]);
    }

    return 0;
}