P4451-生成函数

P4451

题目描述

题解

设斐波那契数列的生成函数为$F(x)=\sum f_i{x}^i$
那么答案$G(x)$的生成函数即为：
$$G(x)=\sum _{i=0}^{\infty }F(x{)}^i=\frac{1}{1-F(x)}$$
考虑怎么求出$F(x)$，这里有一种普遍的方法：
$$\begin{gathered} f_i=f_{i-1}+f_{i-2}+[n=1] \\ F(x)=\sum f_i{x}^i=\sum (f_{i-1}+f_{i-2}+[n=1])x^i \\ F(x)=\sum f_{i-1}{x}^i+\sum f_{i-2}{x}^i+x=xF(x)+x^{2}F(x)+x \\ F(x)=\frac{x}{1-x-x^2} \end{gathered}$$

那么$G(x)$为：
$$G(x)=\frac{x^2+x-1}{x^2+2x-1}=1-\frac{x}{x^2+2x-1}$$
由于$n$过大，不能直接求。
类似这样的问题的实质即是给定有理函数$P(x)=\frac{Q(x}{R(x)}$，求$P(x)$的某项系数，解决的方法通常有待定系数法和不同根的有理展开定理。

这里用待定系数法。
首先设：
$$x^2+2x-1=(1-p_{1}x)(1-p_{2}x)$$
解得$p_1,p_2$
再设：
$$\frac{x}{x^2+2x-1}=\frac{a}{1-p_{1}x}+\frac{b}{1-p_{2}x}$$
解得$a,b$
又因为：
$$[x^n]\frac{a}{1-p_{1}x}=a{p}_1^n$$
那么就可以得到$x^n$的系数了

代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int mod=1e9+7; 
int ksm(int a,int b){
	int ans=1;
	while(b){
		if(b&1) ans=ans*a%mod;
		a=a*a%mod;
		b>>=1;
	}return ans;
}
signed main(){
	int re=0;char ch;
	for(ch=getchar();isdigit(ch);ch=getchar()) re=re*10%(mod-1)+ch-'0';
	re=re-1;
	cout<<(1ll*485071604*ksm(1ll*940286408,re%(mod-1))%mod+1ll*514928404*ksm(1ll*59713601,re%(mod-1))%mod)%mod<<endl;
	return 0;
}