2019.03.29【洛谷P4451】【BZOJ2173】整数的lqp拆分（生成函数）

洛谷传送门

BZOJ传送门

---

解析：

设$g_i$表示一个整数的lqp拆分的权值，其生成函数为$G(x)$，$f_i$表示第$i$个斐波那契数，生成函数为$F(x)$。

令$g_0=0,g_1=1$

很显然我们可以发现：
$$\begin{gathered} g_n=\sum _{i=1}^n{g}_i{f}_{n-i}+f_n \\ G=G\ast F+F \end{gathered}$$

也就是$G=\frac{F}{1-F}$

我们知道斐波那契数列的生成函数闭形式是$F=\frac{1}{1-x-x^2}$，代到上面去可以得到$G=\frac{x}{1-2x-x^2}$。

也就是$G=2xG+x^{2}G+x$，最后面那个$1$是对于$g_1$的，不用管，对于$i>1$，我们得到递推式$g_i=2g_{i-1}+g_{i-2}$。

推到这里已经可以过这道题了，那么我们把$n$放到$1e18$如何？

显然我们继续推是可以得到通项公式的。

拆掉分母：$$\frac{x}{1-2x-x^2}=\frac{x}{(1-(1-\sqrt{2})x)(1-(1+\sqrt{2})x)}$$

待定系数，设$$G=\frac{a}{(1-\sqrt{2})x}+\frac{b}{1-(1+\sqrt{2})x}$$

恒等变形解得$a=-\frac{\sqrt{2}}{4},b=\frac{\sqrt{2}}{4}$

随手写一个二次剩余，或者直接暴力$1e9+7$，也不慢，$10s$内随便跑出来，我们得到$59713600^2\equiv 2(\mathrm{m}\mathrm{o}\mathrm{d}1e9+7)$

通项公式：$$g_n=\frac{\sqrt{2}}{4}((1+\sqrt{2}{)}^n-(1-\sqrt{2}{)}^n)$$

---

代码：（递推）

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

cs int mod=1e9+7;
inline int add(int a,int b){return a+b>=mod?a+b-mod:a+b;}
int n;
int g[1000006];
signed main(){
	std::cin>>n;
	g[1]=1;
	for(int re i=2;i<=n;++i)g[i]=add(add(g[i-1],g[i-1]),g[i-2]);
	std::cout<<g[n]<<"\n";
	return 0;
}

代码：（通项）

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

cs int mod=1e9+7;
inline int add(int a,int b){return a+b>=mod?a+b-mod:a+b;}
inline int dec(int a,int b){return a<b?a-b+mod:a-b;}
inline int mul(int a,int b){return (ll)a*b%mod;}
inline int quickpow(int a,int b,int res=1){
	while(b){
		if(b&1)res=mul(res,a);
		a=mul(a,a);
		b>>=1;
	}
	return res;
}

cs int sqr2=59713600,inv2=(mod+1)>>1;

int n;
signed main(){
	std::cin>>n;
	std::cout<<mul(mul(sqr2,mul(inv2,inv2)),dec(quickpow(add(1,sqr2),n),quickpow(dec(1,sqr2),n)));
	return 0;
}