BZOJ2734 [HNOI2012]集合选数

Address

• 洛谷P3226 BZOJ2734

---

Solution

• 讲真，这题看上去一点不像状压。

• 若 $x$ 在该子集中，则 $2x$ 和 $3x$ 不能在该子集中，考虑构造出一个矩阵：
  

  $$\begin{matrix} 1 & 3 & 9 & 27 & ...\\ 2 & 6 & 18 & 54 & ...\\ 4 & 12 & 36 & 108 & ...\\ 8 & 24 & 72 & 216 & ...\\ ... & ... & ... & ... & ...\\ \end{matrix}$$

• 即左上角的数为1， 矩阵中的每个数都等于它左边的数的3倍，等于它上边的数的两倍。

• 则此问题就转化成了在矩阵中取出一些元素，这些元素互不相邻的方案数。
• 考虑到 $n \le 10^5$，实际上这个矩阵最多有18行，最多有11列，我们就能够用状压DP来解决方案数了。
• 考虑设 $f[i][j]$ 表示选到第 $i$ 行，第 $i$ 行的选取状态为 $j$ 的方案数，则 $f[i][j] = \sum f[i - 1][k](第i - 1行、第 i 行选取元素互不相邻)$（注意要用3倍的变化作为列，能够减少状态数，提高时间效率）。
• 但是我们会发现，5,7这种数并不在这个矩阵中。
• 所以这种矩阵可能有多个，在找完1为左上角的矩阵后，我们只需要寻找1~n中的下一个没有出现在矩阵中的元素充当左上角，再次计算即可。
• 由于不同矩阵中的元素互不影响，所以我们只需要把所有可能矩阵的方案数利用乘法原理乘在一起即可。

Code

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>

using namespace std;

typedef long long ll;
const int N = 1e5 + 5; 
const int R = 20, C = 12;
const int Mod = 1e9 + 1;
int a[R][R], f[R][1 << C], c[R];
bool vis[N]; int n, r, Ans = 1; 

inline int work(int x)
{
    vis[a[1][1] = x] = true; r = 1;
    while (1)
    {
        c[r] = 1; int tmp;
        while (tmp = a[r][c[r]] * 3, tmp <= n && !vis[tmp])
            vis[a[r][++c[r]] = tmp] = true;
        if (tmp = a[r][1] << 1, tmp <= n && !vis[tmp])
            vis[a[++r][1] = tmp] = true;
        else break;
    }

    f[0][0] = 1; c[++r] = 0;
    for (int i = 1; i <= r; ++i)
        for (int j = 0, jm = 1 << c[i]; j < jm; ++j)
            f[i][j] = 0;
    for (int i = 0; i < r; ++i)
        for (int j = 0, jm = 1 << c[i]; j < jm; ++j)
        {
            if ((j >> 1 & j) || !f[i][j]) continue;
            for (int k = 0, km = 1 << c[i + 1]; k < km; ++k)
            {
                if ((k >> 1 & k) || (j & k)) continue;
                (f[i + 1][k] += f[i][j]) %= Mod; 
            }
        }

    return f[r][0];
}

int main()
{
//  freopen("set.in", "r", stdin);
//  freopen("set.out", "w", stdout);

    scanf("%d", &n);
    for (int i = 1; i <= n; ++i)
        if (!vis[i]) Ans = (ll)Ans * work(i) % Mod;
    printf("%d\n", Ans);

//  fclose(stdin); fclose(stdout);
    return 0;
}