CF717A Festival Organization(第一类斯特林数,斐波那契数列)

题目大意:求 $\sum\limits_{n=l}^{r}\dbinom{f_n}{k}\bmod 10^9+7$。其中 $f_n$ 是长度为 $n$ 的 $01$ 序列中,没有连续两个或超过两个 $0$ 的个数。

$1\le k\le 200,1\le l\le r\le 10^{18}$。


先考虑如何求 $f_n$。

令 $g[i][j]$ 表示长度为 $i$,结尾是 $j$ 的序列个数。

$$g[i][0]=g[i-1][1]$$

$$g[i][1]=g[i-1][0]+g[i-1][1]$$

将第一个式子代入第二个式子有 $g[i][1]=g[i-2][1]+g[i-1][1]$。

手玩发现 $g[1][1]=1,g[2][1]=2$。那么有 $g[i][1]=fib_{i+1}$。($fib$ 是斐波那契数列)

要求 $g[i][0]+g[i][1]=g[i-1][1]+g[i][1]=g[i+1][1]=fib_{i+2}$。

那么式子就是 $\sum\limits_{n=l+2}^{r+2}\dbinom{fib_n}{k}$。(为方便 $l+=2,r+=2$,下文假设是 $l$ 到 $r$)

$$\sum\limits_{n=l}^r\dbinom{fib_n}{k}$$

$$\dfrac{1}{k!}\sum\limits_{n=l}^rfib_n^{\underline{k}}$$

$$\dfrac{1}{k!}\sum\limits_{n=l}^r\sum\limits_{i=0}^k(-1)^{k-i}\begin{bmatrix}k\\i\end{bmatrix}fib_n^i$$

$$\dfrac{1}{k!}\sum\limits_{i=0}^k(-1)^{k-i}\begin{bmatrix}k\\i\end{bmatrix}\sum\limits_{n=l}^rfib_n^i$$

前面可以随便枚举,然而斐波那契 $k$ 次方前缀和这东西怎么搞?

这时就要用上大名鼎鼎的通项公式:(没记住的也可以现场用特征方程推)

$$fib_n=\dfrac{\sqrt{5}}{5}(\dfrac{1+\sqrt{5}}{2})^n-\dfrac{\sqrt{5}}{5}(\dfrac{1-\sqrt{5}}{2})^n$$

令 $a=\dfrac{\sqrt{5}}{5},b=-\dfrac{\sqrt{5}}{5},x=\dfrac{1+\sqrt{5}}{2},y=\dfrac{1-\sqrt{5}}{2}$,那么 $fib_n=ax^n+by^n$。

$$\dfrac{1}{k!}\sum\limits_{i=0}^k(-1)^{k-i}\begin{bmatrix}k\\i\end{bmatrix}\sum\limits_{n=l}^r(ax^n+by^n)^i$$

$$\dfrac{1}{k!}\sum\limits_{i=0}^k(-1)^{k-i}\begin{bmatrix}k\\i\end{bmatrix}\sum\limits_{n=l}^r\sum\limits_{j=0}^i\dbinom{i}{j}(ax^n)^j(by^n)^{i-j}$$

$$\dfrac{1}{k!}\sum\limits_{i=0}^k(-1)^{k-i}\begin{bmatrix}k\\i\end{bmatrix}\sum\limits_{n=l}^r\sum\limits_{j=0}^i\dbinom{i}{j}(a^jb^{i-j})(x^jy^{i-j})^n$$

$$\dfrac{1}{k!}\sum\limits_{i=0}^k(-1)^{k-i}\begin{bmatrix}k\\i\end{bmatrix}\sum\limits_{j=0}^i\dbinom{i}{j}(a^jb^{i-j})\sum\limits_{n=l}^r(x^jy^{i-j})^n$$

此时 $i$ 和 $j$ 可以 $O(k^2)$ 枚举,而 $n$ 那里是个等差数列,总复杂度 $O(k^2\log r)$。(可能要预处理出所有斯特林数和组合数才能做到这个复杂度)

但是有个严重的问题:$\sqrt{5}$ 在模 $10^9+7$ 意义下不存在。

那么就要用到一个骚操作:扩系。

令 $(a,b)=a+b\sqrt{5}$,那么上文中的四个常数 $a=(0,5^{-1}),b=(0,-5^{-1}),x=(2^{-1},2^{-1}),y=(2^{-1},-2^{-1})$。

那么进行加减乘除就简单了:

$$(a,b)\pm(c,d)=(a\pm c,b\pm d)$$

$$(a,b)\times(c,d)=(ac+5bd,ad+bc)$$

$$\dfrac{1}{(a,b)}=(\dfrac{a}{a^2-5b^2},\dfrac{-b}{a^2-5b^2})$$

因为式子没推错(只能这么想了啊),最后求出的一定是整数。

那么就做完了。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=222,mod=1000000007,inv2=500000004,inv5=400000003;
#define FOR(i,a,b) for(int i=(a);i<=(b);i++)
#define ROF(i,a,b) for(int i=(a);i>=(b);i--)
#define MEM(x,v) memset(x,v,sizeof(x))
inline ll read(){
    char ch=getchar();ll x=0,f=0;
    while(ch<'0' || ch>'9') f|=ch=='-',ch=getchar();
    while(ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
    return f?-x:x;
}
int k,fac[maxn],invfac[maxn],ans,S[maxn][maxn],C[maxn][maxn];
ll l,r;
inline int add(int x,int y){return x+y<mod?x+y:x+y-mod;}
inline int sub(int x,int y){return x<y?x-y+mod:x-y;}
inline int mul(int x,int y){return 1ll*x*y%mod;}
inline int qpow(int a,ll b){
    int ans=1;
    for(;b;b>>=1,a=mul(a,a)) if(b&1) ans=mul(ans,a);
    return ans;
}
struct comp{
    int x,y;
    comp(const int xx=0,const int yy=0):x(xx),y(yy){}
    inline comp operator+(const comp &c)const{return comp(add(x,c.x),add(y,c.y));}
    inline comp operator-(const comp &c)const{return comp(sub(x,c.x),sub(y,c.y));}
    inline comp operator*(const comp &c)const{return comp(add(mul(x,c.x),mul(5,mul(y,c.y))),add(mul(x,c.y),mul(y,c.x)));}
    inline comp inv()const{
        comp ans(x,y?mod-y:0);
        int dn=qpow(sub(mul(x,x),mul(5,mul(y,y))),mod-2);
        return ans*dn;
    }
    inline comp operator/(const comp &c)const{return *this*c.inv();}
    inline bool operator==(const comp &c)const{return x==c.x && y==c.y;}
}a(0,inv5),b(0,mod-inv5),x(inv2,inv2),y(inv2,mod-inv2);
inline comp cqpow(comp a,ll b){
    comp ans(1,0);
    for(;b;b>>=1,a=a*a) if(b&1) ans=ans*a;
    return ans;
}
comp calc(comp x,ll l,ll r){
    if(x==1) return (r-l+1)%mod;
    return (cqpow(x,r+1)-cqpow(x,l))/(x-1);
}
int main(){
    k=read();l=read();r=read();
    FOR(i,0,k) C[i][0]=C[i][i]=1;
    FOR(i,1,k) FOR(j,1,i-1) C[i][j]=add(C[i-1][j],C[i-1][j-1]);
    S[1][1]=1;
    FOR(i,2,k) FOR(j,1,i) S[i][j]=add(mul(i-1,S[i-1][j]),S[i-1][j-1]);
    fac[0]=1;
    FOR(i,1,k) fac[i]=mul(fac[i-1],i);
    invfac[k]=qpow(fac[k],mod-2);
    FOR(i,0,k){
        int s=0;
        FOR(j,0,i){
            comp tmp1=cqpow(a,j)*cqpow(b,i-j),tmp2=cqpow(x,j)*cqpow(y,i-j);
            s=add(s,mul(C[i][j],(tmp1*calc(tmp2,l+2,r+2)).x));
        }
        s=mul(s,S[k][i]);
        if((k-i)&1) ans=sub(ans,s);
        else ans=add(ans,s);
    }
    printf("%d\n",mul(ans,invfac[k]));
}
View Code

猜你喜欢

转载自www.cnblogs.com/1000Suns/p/11030886.html
今日推荐