BZOJ2734 [HNOI2012]集合选数
Description
求出集合{1,2,…..,n}中,满足:
诺x在该集合中,2x与3x不在集合。
的所有子集的个数。对答案mod 1000000001。
题解
这个题目很要思维了,反正我没想出来,但看了题解十分佩服。
我们构造一个矩阵,
x | 3x | 9x | .. | 3jx |
---|---|---|---|---|
2x | 6x | 18x | .. | .. |
.. | .. | .. | .. | .. |
2ix | .. | .. | .. | 2i3jx |
可以发现,满足要求的子集必须满足一个条件,就是从矩阵中取出的任意两个数不能相邻。因为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;
}