@codeforces - 717A@ Festival Organization

目录

• @description@
• @solution@
• @accepted code@
• @details@

---

@description@

一个长度为 n 的 01 序列是好的，当且仅当该序列任意两个 0 不相邻。

求从 [l, r] 中选出 k 个长度相等的 01 序列的方案数。

1 ≤ k ≤ 200, 1 ≤ l ≤ r ≤ 10^18。

原题戳这里

@solution@

定义 f[i] 表示合法 01 串数量，对最后一个是 1 还是 0 进行讨论得到 f[i] = f[i-1] + f[i-2]。

递推式有斐波那契数列形式，但注意 f[1] = 2 所以不完全是斐波那契，不过可以通过移位变成斐波那契。

以下都把它当作斐波那契数列来解决。

则问题相当于问 \(\sum_{i=l}^{r}C_{f_i}^k = \sum_{i=0}^{r}C_{f_i}^k - \sum_{i=0}^{l-1}C_{f_i}^k\)。

可以通过第一类斯特林数化成 \(\sum_{i=0}^{n}C_{f_i}^k = \frac{1}{k!}\sum_{i=0}^{n}(\sum_{j=0}^{k}S_{k}^{j}*f_{i}^j*(-1)^{k-j})\)。

问题可以等价于求 \(\sum_{i=0}^{n}f_{i}^p\)。当 p = 1, 2 的时候都可以搜到相应的式子。是否 p 增大也有相应的式子呢？

除了把递推式写成矩阵以外，斐波那契还有一个常见解决方法：通项公式。众所周知，斐波那契的通项公式可以写作 \(f[i] = a*A^i + b*B^i\)。

注意通项公式中含有 \(\sqrt{5}\)，而该模数下 5 没有二次剩余。所以需要类似复数，将数存成 \((a + b*\sqrt{5})\) 的形式。

然后直接把通项公式往里面代：

\[\sum_{i=0}^{n}f_{i}^p = \sum_{i=0}^{n}(a*A^i + b*B^i)^p
\\ = \sum_{i=0}^{n}\sum_{j=0}^{p}C_p^j*(a*A^i)^j*(b*B^j)^{p-j}
\\ = \sum_{j=0}^{p}C_p^j*a^j*b^{p-j}\sum_{i=0}^{n}(A^j*B^{p-j})^i\]

等比数列求和解后面的式子即可。

最后复杂度 O(k^2logA)（还有等比数列快速幂的复杂度）。

@accepted code@

#include <cstdio>
#include <algorithm>
using namespace std;

typedef long long ll;

const int MAXK = 200;
const int MOD = int(1E9) + 7;

struct mint{
	int x, y;
	mint(int _x=0, int _y=0) : x(_x), y(_y) {}
	friend mint operator + (mint a, mint b) {
		int x = (a.x + b.x >= MOD ? a.x + b.x - MOD : a.x + b.x);
		int y = (a.y + b.y >= MOD ? a.y + b.y - MOD : a.y + b.y);
		return mint(x, y);
	}
	friend mint operator - (mint a, mint b) {
		int x = (a.x - b.x < 0 ? a.x - b.x + MOD : a.x - b.x);
		int y = (a.y - b.y < 0 ? a.y - b.y + MOD : a.y - b.y);
		return mint(x, y);
	}
	friend mint operator * (mint a, mint b) {
		int x = (1LL*a.x*b.x + 5LL*a.y*b.y) % MOD;
		int y = (1LL*a.x*b.y + 1LL*a.y*b.x) % MOD;
		return mint(x, y);
	}
	friend mint mpow(mint a, ll k) {
		mint r = 1;
		while( k ) {
			if( k & 1 ) r = r * a;
			a = a * a;
			k >>= 1;
		}
		return r;
	}
	friend mint operator / (mint a, mint b) {
		mint p = a * mint(b.x, (MOD - b.y)%MOD), q = mint((1LL*b.x*b.x%MOD + MOD - 5LL*b.y*b.y%MOD)%MOD, 0);
		return p * mpow(q, MOD-2);
	}
}s[MAXK + 5][MAXK + 5], c[MAXK + 5][MAXK + 5];

const mint A1 = 1/mint(0,1);
const mint A2 = (1 + mint(0,1))/2;
const mint B1 = 1/mint(0,MOD-1);
const mint B2 = (1 - mint(0,1))/2;

void init() {
	for(int i=0;i<=MAXK;i++) {
		c[i][0] = 1;
		for(int j=1;j<=i;j++)
			c[i][j] = c[i-1][j] + c[i-1][j-1];
	}
	for(int i=0;i<=MAXK;i++)
		for(int j=0;j<=i;j++)
			c[i][j] = c[i][j] * mpow(A1, j) * mpow(B1, i-j);
	s[0][0] = 1;
	for(int i=1;i<=MAXK;i++)
		for(int j=1;j<=i;j++)
			s[i][j] = s[i-1][j-1] + (i-1)*s[i-1][j];
	for(int i=0;i<=MAXK;i++)
		for(int j=0;j<=MAXK;j++)
			if( (i + j) & 1 ) s[i][j] = 0 - s[i][j];
}

int k;
mint get(mint a, ll m) {
	if( a.x == 1 && a.y == 0 ) return m%MOD + 1;
	else return (mpow(a, m + 1) - 1) / (a - 1);
}
mint solve(ll m) {
	mint ans = 0;
	for(int j=0;j<=k;j++) {
		mint del = 0;
		for(int p=0;p<=j;p++) {
			mint tmp = mpow(A2, p) * mpow(B2, j-p);
			del = del + c[j][p] * get(tmp, m);
		}
		ans = ans + s[k][j] * del;
	}
	return ans;
}

int main() {
	ll l, r; init();
	scanf("%d%lld%lld", &k, &l, &r), l += 2, r += 2;
	mint p = 1; for(int i=1;i<=k;i++) p = p*i;
	printf("%d\n", ((solve(r) - solve(l-1))/p).x);
}

@details@

注意等比数列特判公比为 1 ！！！（老是记不住）

貌似是 BJOI2019 的勘破神机？