Problem - 1152F2 - Codeforces
发现 $k, m$ 很小这个东西肯定有问题。
要么是直接通向要么可以使用矩阵快速幂。
感觉任意两个元素都是不同比较难处理,所以我们考虑进行 $\tt Dp$。
设 $f(i ,j, S)$ 表示当前值是 $i$,是序列长度为 $j$。
因为直接按照序列进行放置值域是不好确定的,我们考虑进行插入操作。
对于值域为 $i$ 我们需要权值 $\ge i - m$ 的位置在其前面,所以 $S$ 表示权值 $\ge i - m$ 的权值有哪些在序列中。
当然我们可以放在序列的开头,因为我们考虑值域是从小到大的,所以可以直接进行操作。
$$
\begin{aligned}
f(i, j, S) \times (popcount(S) + 1) &\to
f(i + 1, j + 1, (2S + 1)) \\
f(i, j, S) &\to f(i + 1, j + 1, (2S))
\end{aligned}
$$
直接矩阵快速幂即可。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79
| #include <bits/stdc++.h> using namespace std; namespace Legendgod { namespace Read {
#ifdef Fread const int Siz = (1 << 21) + 5; char *iS, *iT, buf[Siz]; #define gc() ( iS == iT ? (iT = (iS = buf) + fread(buf, 1, Siz, stdin), iS == iT ? EOF : *iS ++) : *iS ++ ) #define getchar gc #endif template <typename T> void r1(T &x) { x = 0; char c(getchar()); int f(1); for(; !isdigit(c); c = getchar()) if(c == '-') f = -1; for(; isdigit(c); c = getchar()) x = (x << 1) + (x << 3) + (c ^ 48); x *= f; } template <typename T, typename...Args> void r1(T &x, Args&...arg) { r1(x), r1(arg...); } #undef getchar }
using namespace Read;
const int maxn = 200 + 5; constexpr int mod = 1e9 + 7; int n, m, Up, K;
struct Matrix { int a[maxn][maxn]; Matrix(void) { memset(a, 0, sizeof(a)); } Matrix operator * (const Matrix &z) const { Matrix res; for(int i = 0; i <= Up; ++ i) { for(int j = 0; j <= Up; ++ j) { long long tmp(0); for(int k = 0; k <= Up; ++ k) { tmp += 1ll * a[i][k] * z.a[k][j] % mod; } res.a[i][j] = (res.a[i][j] + tmp) % mod; } } return res; } }trans, F;
signed main() { int i, j; r1(n, K, m); Up = (K << m); for(i = 0; i < K; ++ i) for(j = 0; j < (1 << m); ++ j) { int tp = ( (j << 1) & ( (1 << m) - 1) ); int c = 1 + __builtin_popcount(j); trans.a[(i << m) + j][(i << m) + tp] = 1; if(i == K - 1) trans.a[(i << m) + j][Up] = c; else trans.a[(i << m) + j][((i + 1) << m) + tp + 1] = c; }
trans.a[Up][Up] = F.a[0][0] = 1; while(n) { if(n & 1) F = F * trans; trans = trans * trans; n >>= 1; } printf("%d\n", F.a[0][Up]); return 0; }
}
signed main() { return Legendgod::main(), 0; }
|