ZOJ-3931 Exact Compression（DP+bitset）

题意

文章中 $n$ 种单词，第 $i$ 种单词出现的次数为 $F_i$ ，现在要对这些单词进行哈夫曼编码，求文章编码后 $0$ 的出现次数能否等于 $E$ ？
$1 \leq n \leq 128$
$1 \leq E \leq 10^8$
$1 \leq F_i \leq 1000$

思路

哈夫曼编码的执行过程如下：
（1）从优先队列中取出出现次数（元素权值）最少的元素；
（2）将它们的编码前分别增加 $0$ 和 $1$；
（3）将它们出现次数之和插入优先队列。
（4）重复以上操作直至优先队列中只有一个元素。
由这个过程不难发现，将两个元素合并之后，这个对这个新元素的操作就相当于在把它包含着的单词上操作。所以合并后，在这个元素的编码前添加一个零，就相当于在元素中的所有单词编码前添加一个零，文章一共增添了此元素权值个零。
对于一些单词，哈夫曼编码的编码顺序是唯一确定的，那么我们的决策就是在取出两个元素后，在哪个元素前添加 $0$ ，哪个元素前添加 $1$ 。我们用一个 $bitset$ 类型的 $dp$ 维护此时文章中可能出现的 $0$ 的个数。现在取出了权值分别为 $a,b$ 的两个元素，那么文章中的 $0$ 可能会添加 $a$ 个或 $b$ 个，不难看出 $dp=(dp<<a)|(dp<<b)$，于是只需在合并的过程中执行这些转移即可，见代码一。
我们还可以令上述的 $a$ 为先取出的元素， $b$ 为后取出的元素，那么 $b>a$ 。可以把 $b=a+(b-a)$ 那么就可以将所求的 $E$ 减去操作次数个 $a$ 然后把 $b-a$ 看作 $01$ 背包中的物品价值跑 $01$ 背包即可，见代码二。
经检验， $E$ 值的最大值在 $128$ 个不同单词各出现 $1000$ 此时取到，由此得到 $bitset$ 的范围。

代码

代码一

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<bitset>
#include<queue>
#define FOR(i,x,y) for(int i=(x);i<=(y);i++)
#define DOR(i,x,y) for(int i=(x);i>=(y);i--)
#define N 448001
typedef long long LL;
using namespace std;
priority_queue<int,vector<int>,greater<int> >q;
bitset<N>dp;

int main()
{
    int T,n,E,p;
    scanf("%d",&T);
    while(T--)
    {
        dp.reset();
        dp[0]=1;
        while(!q.empty())q.pop();
        scanf("%d",&n);
        FOR(i,1,n)
        {
            scanf("%d",&p);
            q.push(p);
        }
        scanf("%d",&E);
        if(E>=N)
        {
            printf("No\n");
            continue;
        }
        FOR(i,2,n)
        {
            int a=q.top();q.pop();
            int b=q.top();q.pop();
            q.push(a+b);
            dp=(dp<<a)|(dp<<b);
        }
        printf(dp[E]?"Yes\n":"No\n");
    }
    return 0;
}

代码二

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<bitset>
#include<queue>
#define FOR(i,x,y) for(int i=(x);i<=(y);i++)
#define DOR(i,x,y) for(int i=(x);i>=(y);i--)
#define N 448001
typedef long long LL;
using namespace std;
bitset<N>dp;
priority_queue<int,vector<int>,greater<int> >q;
int c[130];

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        while(!q.empty())q.pop();
        dp=1;
        int n,E,p;
        scanf("%d",&n);
        FOR(i,1,n)
        {
            scanf("%d",&p);
            q.push(p);
        }
        scanf("%d",&E);
        if(E>=N)
        {
            printf("No\n");
            continue;
        }
        FOR(i,1,n-1)
        {
            int a=q.top();q.pop();
            int b=q.top();q.pop();
            q.push(a+b);
            dp|=dp<<b-a;
            E-=a;
        }
        if(E<0)printf("No\n");
        else printf(dp[E]?"Yes\n":"No\n");
    }
    return 0;
}