Lightoj 1127 - Funny Knapsack [2 점]

제목 링크:http://www.lightoj.com/volume_showproblem.php?problem=1127
제목: n 개의 물체 (n < 30) 와 1 개의 용량 이 W 인 용기 가 있 습 니 다. 용기 에 가득 채 우지 않 은 물건 을 방치 하 는 방식 이 몇 가지 가 있 는 지 물 어보 세 요.
사고방식: 상태 압축 + 2 점.앞의 n / 2 개의 물 체 를 하나의 전체 로 보고 나머지 는 하나의 전체 로 본다.1 < (n / 2) 개 상 태 는 앞의 절반 의 아 이 템 사용 상황 을 대표 하고 각 상태의 총 부 피 를 구 합 니 다.정렬뒤의 반 에 도.답 은 반 만 매 거 하고 나머지 반 에서 W 와 떨 어 지 는 하 계 를 찾 으 면 된다.
코드:
#include <stdio.h>
#include <ctime>
#include <math.h>
#include <limits.h>
#include <complex>
#include <string>
#include <functional>
#include <iterator>
#include <algorithm>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <list>
#include <bitset>
#include <sstream>
#include <iomanip>
#include <fstream>
#include <iostream>
#include <ctime>
#include <cmath>
#include <cstring>
#include <cstdio>
#include <time.h>
#include <ctype.h>
#include <string.h>
#include <assert.h>

using namespace std;

int n;
long long w;
long long a[1000000];
long long s1[1000000];
long long s2[1000000];

int main()
{
    int t;
    scanf("%d", &t);
    int cases = 1;
    while (t--)
    {
        memset(s1, 0, sizeof(s1));
        memset(s2, 0, sizeof(s2));
        memset(a, 0, sizeof(a));

        scanf("%d %lld", &n, &w);
        for (int i = 0; i < n; i++)
            scanf("%lld", &a[i]);

        int t1 = n >> 1;
        int t2 = n - t1;
        int r1 = 1 << t1;
        int r2 = 1 << t2;

        for (int i = 0; i < r1; i++)
            for (int j = 0; j < t1; j++)
            {
                if (i &(1 << j))
                    s1[i] += a[j];
            }

        for (int i = 0; i < r2; i++)
            for (int j = 0; j < t2; j++)
            {
                if (i &(1 << j))
                    s2[i] += a[t1 + j];
            }

        sort(s2, s2 + r2 );

        long long ans = 0;
        for (int i = 0; i < r1; i++)
        {
            if (w - s1[i] >= 0)
                ans += upper_bound(s2, s2 + r2, w - s1[i]) - s2;
        }
        printf("Case %d: %lld
"
, cases++, ans); } return 0; }

좋은 웹페이지 즐겨찾기