【bzoj2734】[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}。

题解

思路题。。我反正是想不出。
http://hzwer.com/5149.html
考虑列出一个这样的矩阵

$$\begin{bmatrix} 1 & 3 & 9 & 27\dots\\ 2 & 6 & 18 & 54\dots\\ 4 & 12 & 36 & 108\dots \end{bmatrix}$$
这个矩阵不会很大，20行20列以内。
而我们取数的限制就是在这个矩阵中相邻的两个数不能同时取。
那么显然可以把一行压成一个二进制数，然后状压DP解决。
注意到一个矩阵不可能包含 $1\dots n$的所有数，那么我们就标记已经出现过的数，对于没出现过的数将它作为矩阵的第一行第一列的元素再计算一遍，因为不同矩阵的答案互不干扰，所以用乘法原理相乘起来即可。

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;

const int N = 100000 + 10, mod = 1000000001, M = 20;

int n, vis[N], a[M][M], R[M];
ll ans = 1, f[M][N];

ll cal(int x){
    memset(R, 0, sizeof(R));
    a[1][1] = x;
    int C;
    for(int i = 1; i < M; i++){
        if(i != 1){
            a[i][1] = a[i-1][1] * 2;
            if(a[i][1] > n){
                C = i - 1;
                break;
            }
        }
        vis[a[i][1]] = 1;
        for(int j = 2; j < M; j++){
            a[i][j] = a[i][j-1] * 3;
            if(a[i][j] > n){
                R[i] = j - 1;
                break;
            }
            vis[a[i][j]] = 1;
        }
    }
    for(int i = 0; i <= C; i++) for(int j = 0; j < (1<<R[i]); j++) f[i][j] = 0;
    f[C+1][0] = 0;
    f[0][0] = 1;
    for(int i = 0; i <= C; i++){
        for(int j = 0; j < (1<<R[i]); j++){
            if(f[i][j]){
                if(j & (j >> 1)) continue;
                for(int k = 0; k < (1<<R[i+1]); k++)
                    if(!(j & k) && !(k & (k >> 1))) f[i+1][k] = (f[i+1][k] + f[i][j]) % mod;
            }
        }
    }
    return f[C+1][0];
}

void work(){
    scanf("%d", &n);
    for(int i = 1; i <= n; i++)
        if(!vis[i]) ans = (ans * cal(i)) % mod;
    printf("%d\n", ans);
}

int main(){
    work();
    return 0;
}