一、问题分析
将问题转化为:一个
N
个数的序列,序列的每个位置需要填充一个
二、DP模型
分析后可以得出,由于前
K
个填充的数已经确定,所以这里只关心序列中的数相同或不相同,而和具体数值无关。因此设
三、DP转移
为了保证每个位置都能被填充,规定在转移过程中,
S
的第
而判断
S2
能否从
S1
转移,就是把
S1
最高位的
1
去掉之后在末尾再补一个
四、矩阵优化
考虑到
N≤109
的数据范围,想到矩阵乘法。这时根据上面推出的转移条件,得到有用的状态只有
CK−1P−1
个。可以得到矩阵规模的最大值为
C⌊92⌋9=126
。
复杂度
O((CK−1P−1)3logN)
。
五、代码
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 134, ZZQ = 30031;
int n, K, P, tot, sta[N];
struct cyx {
int n, m, a[N][N];
cyx() {}
cyx(int _n, int _m) :
n(_n), m(_m) {memset(a, 0, sizeof(a));}
friend inline cyx operator * (cyx a, cyx b) {
int i, j, k; cyx res = cyx(a.n, b.m);
for (i = 1; i <= res.n; i++) for (j = 1; j <= res.m; j++)
for (k = 1; k <= a.m; k++)
(res.a[i][j] += a.a[i][k] * b.a[k][j]) %= ZZQ;
return res;
}
friend inline cyx operator ^ (cyx a, int b) {
int i; cyx res = cyx(a.n, a.m);
for (i = 1; i <= res.n; i++) res.a[i][i] = 1;
while (b) {
if (b & 1) res = res * a;
a = a * a;
b >>= 1;
}
return res;
}
} Orz, Zzq;
int main() {
int i, j, k; cin >> n >> K >> P;
for (i = (1 << P - 1); i < (1 << P); i++) {
int cnt = 0; for (j = 0; j < P; j++) if ((i >> j) & 1) cnt++;
if (cnt == K) sta[++tot] = i;
}
Orz = cyx(tot, tot); Zzq = cyx(tot, 1);
for (i = 1; i <= tot; i++) for (j = 1; j <= tot; j++) {
int S1 = sta[i], S2 = sta[j]; S1 = S1 - (1 << P - 1) << 1;
for (k = 0; k < P; k++) if (!((S1 >> k) & 1) && S1 + (1 << k) == S2)
Orz.a[j][i] = 1;
}
Zzq.a[tot][1] = 1; Orz = (Orz ^ n - K) * Zzq;
cout << Orz.a[tot][1] << endl;
return 0;
}