[CQOI 2018]交错序列

时间:2021-02-07 19:24:22

Description

题库链接

定义长度为 \(n\) 的“交错序列”为:长度为 \(n\) 序列中仅含 \(0,1\) 且没有相邻的 \(1\) 。给出 \(a,b\) ,假设序列中有 \(x\) 个 \(0\) , \(y\) 个 \(1\) 。定义该“交错序列”的价值为 \(x^a\times y^b\) ,特别地 \(0^0=1\) 。求所有长度为 \(n\) 的交错序列的价值的和。对质数 \(m\) 取模。

\(1\leq n\leq 10000000,1\leq m\leq 100000000,1\leq a,b\leq 45\)

Solution

傻逼题啊,组合数+ \(lucas\) 乱搞一下就好了。

式子就是: \(\sum\limits_{i=0}^{\left\lfloor\frac{n+1}{2}\right\rfloor}(n-i)^a\times i^b\times{n-i+1\choose i}\)

正好 \(O(n)\) ,乱搞交了

あら? \(lucas\) 也有个 \(\log\) 啊!!! f**k , 45pts 。

观察式子 \(x^a\times y^b\) ,等价于 \((n-y)^a\times y^b\) 。

\[\sum_{i=0}^a{a\choose i}(-1)^{a-i}n^iy^{a+b-i}\]

那么我们只要算出 \(y^i\) ,最后乘上 \({a\choose i}(-1)^{a-i}n^i\) 即可。

记 \(f_{i,j,0/1}\) 表示长度为 \(i\) 的序列结尾为 \(0/1\) 且所有 \(1\) 的个数(即式子中的 \(y\) )的 \(j\) 次方和。

考虑转移,注意到如果这一位填 \(0\) ,那么是与式子中的 \(y\) 无关的,显然

\[f_{i-1,j,0}+f_{i-1,j,1}\rightarrow f_{i,j,0}\]

如果这一位填 \(1\) ,假设 \(y'=y+1\) ,那么 \(y'^i=(y+1)^i=\sum\limits_{j=0}^i {i\choose j}y^j\)

那么,

\[\sum_{k=0}^j{j\choose k}f_{i-1,k,0}\rightarrow f_{i,j,1}\]

初值 \(f_{0,0,0}=1\) ,长度为 \(0\) 含有 \(0\) 个 \(1\) 。可以用矩阵乘法加速。

复杂度 \(O((2(a+b))^3\log_2 n)\) 。有点卡常...

Code

45pts

#include <bits/stdc++.h>
using namespace std;
const int N = 10000000+5; int n, m, a, b, ifac[N], fac[N]; int quick_pow(int a, int b) {
int ans = 1;
while (b) {
if (b&1) ans = 1ll*ans*a%m;
b >>= 1, a = 1ll*a*a%m;
}
return ans;
}
int C(int n, int m, int p) {
if (n < m) return 0;
return 1ll*fac[n]*ifac[m]%p*ifac[n-m]%p;
}
int lucas(int n, int m, int p) {
if (!m || n == m) return 1; if (n < m) return 0;
return 1ll*lucas(n/p, n/p, p)*C(n%p, m%p, p)%p;
}
void work() {
scanf("%d%d%d%d", &n, &a, &b, &m);
int l = min(n+1, m);
ifac[0] = ifac[1] = fac[0] = fac[1] = 1;
for (int i = 2; i < l; i++) ifac[i] = -1ll*(m/i)*ifac[m%i]%m;
for (int i = 2; i <= l; i++)
ifac[i] = 1ll*ifac[i]*ifac[i-1]%m,
fac[i] = 1ll*i*fac[i-1]%m;
int ans = 0;
for (int i = 0; i*2-1 <= n; i++) {
(ans += 1ll*quick_pow(n-i, a)*quick_pow(i, b)%m*lucas(n-i+1, i, m)%m) %= m;
}
printf("%d\n", (ans+m)%m);
}
int main() {work(); return 0; }

100pts

#include <bits/stdc++.h>
using namespace std;
const int N = 200+5; int n, a, b, m, C[N][N], sz1, sz2;
struct mat {
int a[N][N];
mat() {}
mat operator * (const mat &b) const {
mat ans;
for (int i = 0; i < sz2; i++)
for (int j = 0; j < sz2; j++) {
ans.a[i][j] = 0;
for (int k = 0; k < sz2; k++)
if (a[i][k] && b.a[k][j])
(ans.a[i][j] += 1ll*a[i][k]*b.a[k][j]%m) %= m;
}
return ans;
}
}S, T; void work() {
scanf("%d%d%d%d", &n, &a, &b, &m);
sz1 = a+b+1, sz2 = sz1<<1;
for (int i = 0; i <= sz1; i++) {
C[i][0] = 1;
for (int j = 1; j <= i; j++) C[i][j] = (C[i-1][j-1]+C[i-1][j])%m;
}
for (int i = 0; i < sz1; i++) {
T.a[i][i] = 1, T.a[i+sz1][i] = 1;
for (int j = i; j < sz1; j++) T.a[i][j+sz1] = C[j][i];
}
S.a[0][0] = 1; int t = n;
while (t) {
if (t&1) S = S*T;
T = T*T, t >>= 1;
}
int ans = 0;
for (int i = 0, pw = 1; i <= a; i++, pw = 1ll*pw*n%m) {
ans += 1ll*((a-i)&1 ? -1 : 1)*C[a][i]*pw%m*(S.a[0][a+b-i]+S.a[0][a+b+sz1-i])%m;
ans %= m;
}
printf("%d\n", (ans+m)%m);
}
int main() {work(); return 0; }