【刷题】BZOJ 2734 [HNOI2012]集合选数

Description

《集合论与图论》这门课程有一道作业题，要求同学们求出{1, 2, 3, 4, 5}的所有满足以 下条件的子集：若 x 在该子集中，则 2x 和 3x 不能在该子集中。同学们不喜欢这种具有枚举性 质的题目，于是把它变成了以下问题：对于任意一个正整数 n≤100000，如何求出{1, 2,..., n} 的满足上述约束条件的子集的个数（只需输出对 1,000,000,001 取模的结果），现在这个问题就 交给你了。

Input

只有一行，其中有一个正整数 n，30%的数据满足 n≤20。

Output

仅包含一个正整数，表示{1, 2,..., n}有多少个满足上述约束条件 的子集。

Sample Input

4

Sample Output

8

【样例解释】

有8 个集合满足要求，分别是空集，{1}，{1，4}，{2}，{2，3}，{3}，{3，4}，{4}。

Solution

考虑构造这样一个矩阵
\[ \begin{bmatrix} x & 3x & 9x & 27x & \cdots & \\ \\ 2x & 6x & 18x & 54x & \cdots & \\ \\ 4x & 12x & 36x & 108x & \cdots & \\ \\ 8x & 24x & 72x & 216x & \cdots & \\ \\ \vdots & \vdots & \vdots & \vdots & \ddots &\\ \end{bmatrix} \]
那么对于一个数 \(x\) ，与它有冲突关系的数就都选出来了，我们只要在矩阵上选数，并且相邻位置不能选到就可以了
这一步就是个状压dp模板题
对于不在同一矩阵的每一种 \(x\) ，都构造一个矩阵，计算答案。这些矩阵之间互不干扰，所以直接乘起来就好了

#include<bits/stdc++.h>
#define ui unsigned int
#define ll long long
#define db double
#define ld long double
#define ull unsigned long long
const int MAXN=100000+10,Mod=1e9+1;
int n,G[20],cnt,p[MAXN],vis[MAXN];
ll f[20][2000],ans;
template<typename T> inline void read(T &x)
{
    T data=0,w=1;
    char ch=0;
    while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
    if(ch=='-')w=-1,ch=getchar();
    while(ch>='0'&&ch<='9')data=((T)data<<3)+((T)data<<1)+(ch^'0'),ch=getchar();
    x=data*w;
}
template<typename T> inline void write(T x,char ch='\0')
{
    if(x<0)putchar('-'),x=-x;
    if(x>9)write(x/10);
    putchar(x%10+'0');
    if(ch!='\0')putchar(ch);
}
template<typename T> inline void chkmin(T &x,T y){x=(y<x?y:x);}
template<typename T> inline void chkmax(T &x,T y){x=(y>x?y:x);}
template<typename T> inline T min(T x,T y){return x<y?x:y;}
template<typename T> inline T max(T x,T y){return x>y?x:y;}
inline ll qexp(ll a,ll b)
{
    ll res=1;
    while(b)
    {
        if(b&1)res=res*a%Mod;
        a=a*a%Mod;
        b>>=1;
    }
    return res;
}
inline bool check(int st)
{
    return (st&(st<<1))||(st&(st>>1));
}
inline ll solve(ll x)
{
    memset(G,0,sizeof(G));
    int w=0,h=0;
    while(x*qexp(2,w)<=n)++w;
    while(x*qexp(3,h)<=n)++h;
    for(register int i=1;i<=w;++i)
        for(register int j=1;j<=h;++j)
            if(qexp(2,i-1)*qexp(3,j-1)*x<=n)vis[qexp(2,i-1)*qexp(3,j-1)*x]=1,G[i]|=(1<<h-j);
    cnt=0;
    for(register int st=0;st<(1<<h);++st)
        if(!check(st))p[++cnt]=st;
    memset(f,0,sizeof(f));
    for(register int i=1;i<=w;++i)
        for(register int j=1;j<=cnt;++j)
            if((p[j]|G[i])==G[i])
            {
                if(i==1)f[i][j]++;
                else
                    for(register int k=1;k<=cnt;++k)
                        if((p[k]|G[i-1])==G[i-1]&&!(p[j]&p[k]))(f[i][j]+=f[i-1][k])%=Mod;
            }
    ll res=0;
    for(register int i=1;i<=cnt;++i)
        if((p[i]|G[w])==G[w])(res+=f[w][i])%=Mod;
    return res;
}
int main()
{
    read(n);ans=1;
    for(register int i=1;i<=n;++i)
        if(!vis[i])ans=1ll*ans*solve(i)%Mod;
    write(ans,'\n');
    return 0;
}

转载于:https://www.cnblogs.com/hongyj/p/9517706.html