bzoj 2004 [Hnoi2010]Bus 公交线路 状压dp+矩阵乘法

题面

题目传送门

解法

挺不错的一道状压dp和矩阵乘法的综合题了吧

• 将题目转化一下，就变成有$n$个数，每一个位置都可以填$1-k$这$k$个数，必须保证每一段长度为$p$的子区间中这$k$个数都出现过
• 先观察数据范围，发现$n≤10^9$，$k≤p≤10$，感觉对于$n$要矩乘，对于$k,p$要状压
• 那么我们不妨设$f_{i,j}$表示前$i-1$个数已经填充完毕，$[i,i+p-1]$这$p$个数的取值状态为$j$的方案数
• 为了保证每一个位置都可以被填充，就必须满足$j$的第一位为$1$，且$j$中有且仅有$k$个$1$
• 然后考虑如何转移，假设状态$j'$可以转移到$j$，那么考虑$j'$应该满足什么条件
• 显然，就是$j'$去掉最高位，然后在最后补一个$0$，如果$j'$中的$k-1$个$1$能恰好和$j$中的$k-1$个$1$对应，那么就说明$j'$可以转移到$j$
• 然而我们却发现如果这样的话，状态数为$2^{p-1}=512$，矩乘的话会T
• 其实，真正能在转移中出现的状态数不会超过$x={p-1\choose k-1}$个，最大为${9\choose 4}=126$
• 所以这样就可以实现矩乘了，我们只要把那些会出现过的状态重新标一个号就可以了
• 时间复杂度：$O(x^3\log n)$

代码

#include <bits/stdc++.h>
#define Mod 30031
#define N 150
using namespace std;
template <typename node> void chkmax(node &x, node y) {x = max(x, y);}
template <typename node> void chkmin(node &x, node y) {x = min(x, y);}
template <typename node> void read(node &x) {
    x = 0; int f = 1; char c = getchar();
    while (!isdigit(c)) {if (c == '-') f = -1; c = getchar();}
    while (isdigit(c)) x = x * 10 + c - '0', c = getchar(); x *= f;
}
struct Matrix {
    int a[N][N];
    void Clear() {memset(a, 0, sizeof(a));}
};
int n, k, p, len, s[N];
Matrix operator * (Matrix x, Matrix y) {
    Matrix ret; ret.Clear();
    for (int k = 1; k <= len; k++)
        for (int i = 1; i <= len; i++)
            for (int j = 1; j <= len; j++)
                ret.a[i][j] = (ret.a[i][j] + x.a[i][k] * y.a[k][j] % Mod) % Mod;
    return ret;
}
Matrix operator ^ (Matrix x, int y) {
    Matrix ret = x; y--;
    while (y) {
        if (y & 1) ret = ret * x;
        y >>= 1, x = x * x;
    }
    return ret;
}
int lowbit(int x) {return x & -x;}
void getnum(int dep, int now, int las) {
    if (dep == k) {s[++len] = now; return;}
    for (int i = las - 1; i; i--)
        getnum(dep + 1, now + (1 << i - 1), i);
}
int main() {
    read(n), read(k), read(p);
    getnum(1, 1 << p - 1, p);
    Matrix tx; tx.Clear();
    for (int i = 1; i <= len; i++) cout << s[i] << ' ';
    cout << "\n";
    for (int i = 1; i <= len; i++)
        for (int j = 1; j <= len; j++) {
            int x = (s[i] << 1) ^ (1 << p) ^ s[j];
            if (lowbit(x) && x == lowbit(x)) tx.a[i][j] = 1;
        }
    tx = tx ^ (n - k);
    Matrix ans; ans.Clear(); ans.a[1][1] = 1;
    ans = ans * tx; cout << ans.a[1][1] << "\n";
    return 0;
}