BZOJ2734 [HNOI2012]集合选数

BZOJ2734 [HNOI2012]集合选数

Description

求出集合{1,2,…..,n}中，满足：
诺x在该集合中，2x与3x不在集合。
的所有子集的个数。对答案mod 1000000001。

题解

这个题目很要思维了，反正我没想出来，但看了题解十分佩服。
我们构造一个矩阵，

x	3x	9x	..	$3^jx$
2x	6x	18x	..	..
..	..	..	..	..
$2^ix$	..	..	..	$2^i3^jx$

可以发现，满足要求的子集必须满足一个条件，就是从矩阵中取出的任意两个数不能相邻。因为n<=100000，所以矩阵是大约18×11的，所以我们可以用状态压缩的dp来解决。
f[i][j]为到第1行，第i行状态为j（二进制）中可行的子集数量,那么
f[i][j]+=f[i-1][k]; (j&k)=0&&(j&(j>>1))=0
f[0][0]=1;
其中j&k=0代表上下两行没有相邻的元素，(j&(j>>1))=0表示第i行没有左右相邻的元素。
就这样，可以求出一个矩阵的所以情况。
可是，x为1时，并不能包含所以1到n的数，我们需要枚举没被包含再内的数作为x重新构造矩阵，将所以矩阵的答案乘起来。

#include <cstdio>
#include <cmath>
#include <queue>
#include <algorithm>
#include <cstring>
#define MAXN 100000+10
#define LL long long
#define MOD 1000000001
using namespace std;

int vis[MAXN],n,a[20][20],b[20],f[20][MAXN];
LL ans;
int cal(int x)
{
    memset(b,0,sizeof(b));
    a[1][1]=x;
    for(int i=2;i<=18;i++) 
        if(a[i-1][1]*2<=n) a[i][1]=a[i-1][1]*2;
        else a[i][1]=n+1;

    for(int i=1;i<=18;i++)
        for(int j=2;j<=11;j++)
        if(a[i][j-1]*3<=n) a[i][j]=a[i][j-1]*3;
        else a[i][j]=n+1;

    for(int i=1;i<=18;i++)
        for(int j=1;j<=11;j++)
            if(a[i][j]<=n)
                b[i]+=(1<<(j-1)),vis[a[i][j]]=1; \\b为每一行的状态(二进制)边界

    for(int i=0;i<=18;i++)
        for(int j=0;j<=b[i];j++)
            f[i][j]=0;

    f[0][0]=1;
    for(int i=0;i<18;i++)
        for(int j=0;j<=b[i];j++)
            if(f[i][j])
                for(int k=0;k<=b[i+1];k++)
                    if(((j&k)==0)&&((k&(k>>1))==0))
                        f[i+1][k]=(f[i][j]+f[i+1][k])%MOD;

    return f[18][0];
}
int main()
{
    scanf("%d",&n);
    ans=1;
    for(int i=1;i<=n;i++)
        if(!vis[i]) ans=(1LL*ans*cal(i))%MOD;

    printf("%lld\n",ans);

    return 0;
}