Codeforces Round #554 (Div. 2) F2. Neko Rules the Catniverse (Large Version) （矩阵快速幂 状压DP）

题意

有$n$个点，每个点只能走到编号在$[1,min(n+m,1)]$范围内的点。求路径长度恰好为$k$的简单路径（一个点最多走一次）数。
$1le nle 10^9,1le mle 4,1le kle min(n,12)$

分析

直接考虑走路径的话不能判有没有走过，然后就把路径转化为一个序列，每次往里面插入新的点（神了）。因为一个点可以走到比他小的所有点，那么我们把点从大到小插入。

假设现在已有序列为$p_1,p_2,p_3,...,p_k$。那么当前插入一个点$i$。

假设插在$p_j$和$p_{j+1}$之间，必须满足$p_j$能走到$i$并且$i$能走到$p_{j+1}$。由于$i$是最小的，那么所有$p_j$都能走到$i$，所以只用考虑$i$能走到哪些点。

• 一种情况是直接放在最后。
• 另一种情况是$i+mge p_{j+1}$。那么满足这个式子的$p_{j+1}$最多有$m(le 4)$个。那么就把$[i+1,i+m]$这$m$个数有没有出现在序列中过状压成$m$位$2$进制数，记为$S$。当前方案就是$bitcount(S)$（$S$在$2$进制中有多少个$1$）。

那么一共就有$bitcount(S)+1$种方案。

另外还可以不插入。

所以$DP$状态设为$f[i][j][S]$表示到$i$这个点，序列长度为$j$，上述状态为$S$的方案数，转移方程为：
$egin{aligned} f[i+1][j+1][(S<<1|1)&(2^m-1)]&+=f[i][j][S]*(bitcount(S)+1)\ f[i+1][j][(S<<1|0)&(2^m-1)]&+=f[i][j][S]end{aligned}$

由于$n$比较大，就矩阵加速就行了。这个矩阵快速幂还是挺好写的。。

CODE

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 100005;
const int mod = 1e9 + 7;
int n, k, m, all;
inline void add(int &x, int y) { x += y; if(x >= mod) x -= mod; }
struct mat {
	int a[210][210]; //最大状态数为(k+1)*(1<<m)<=(12+1)*(2^4)=208
	mat() { memset(a, 0, sizeof a); }
	inline mat operator *(const mat &o)const {
		mat re;
		for(int k = 0; k < all; ++k)
			for(int i = 0; i < all; ++i) if(a[i][k])
				for(int j = 0; j < all; ++j) if(o.a[k][j])
					add(re.a[i][j], 1ll * a[i][k] * o.a[k][j] % mod);
		return re;
	}
	inline mat operator ^(int b)const {
		mat re, A = *this;
		for(int i = 0; i < all; ++i) re.a[i][i] = 1;
		while(b) {
			if(b & 1) re = re * A;
			A = A * A; b >>= 1;
		}
		return re;
	}
};
inline int enc(int K, int S) { return K*(1<<m) + S; }
inline int nxt(int S, bool x) { return ((S<<1)|x) & ((1<<m)-1); }
int cnt[16];
int main () {
	scanf("%d%d%d", &n, &k, &m);
	all = (k+1)*(1<<m); //所有状态数
	mat trans, ans;
	ans.a[0][enc(0, 0)] = 1;
	for(int s = 1; s < (1<<m); ++s) cnt[s] = cnt[s>>1] + (s&1); //预处理2进制下有多少个1
	for(int i = 0; i <= k; ++i)
		for(int s = 0; s < (1<<m); ++s) {
			if(i < k) trans.a[enc(i, s)][enc(i+1, nxt(s, 1))] = cnt[s]+1;
			trans.a[enc(i, s)][enc(i, nxt(s, 0))] = 1;
		}
	ans = ans * (trans ^ n);
	int Ans = 0;
	for(int s = 0; s < (1<<m); ++s)
		add(Ans, ans.a[0][enc(k, s)]);
	printf("%d
", (Ans + mod) % mod);
}