洛谷传送门
BZOJ传送门
题目描述
lqp在为出题而烦恼,他完全没有头绪,好烦啊…
他首先想到了整数拆分。整数拆分是个很有趣的问题。给你一个正整数 N N N,对于 N N N的一个整数拆分就是满足任意 m > 0 m>0 m>0, a 1 , a 2 , a 3 … a m > 0 a_1 ,a_2 ,a_3…a_m>0 a1,a2,a3…am>0,且 a 1 + a 2 + a 3 + … + a m = N a_1+a_2+a_3+…+a_m=N a1+a2+a3+…+am=N的一个有序集合。通过长时间的研究我们发现了计算对于 N N N的整数拆分的总数有一个很简单的递推式,但是因为这个递推式实在太简单了,如果出这样的题目,大家会对比赛毫无兴趣的。
然后lqp又想到了斐波那契数。定义 F 0 = 0 , F 1 = 1 , F n = F n − 1 + F n − 2 ( n > 1 ) F_0=0,F_1=1,F_n=F_n-1+F_n-2 (n>1) F0=0,F1=1,Fn=Fn−1+Fn−2(n>1), F n F_n Fn就是斐波那契数的第 n n n项。但是求出第 n n n项斐波那契数似乎也不怎么困难… lqp为了增加选手们比赛的欲望,于是绞尽脑汁,想出了一个有趣的整数拆分,我们暂且叫它:整数的lqp拆分。
和一般的整数拆分一样,整数的lqp拆分是满足任意 m > 0 m>0 m>0, a 1 , a 2 , a 3 … a m > 0 a_1 ,a_2 ,a_3…a_m>0 a1,a2,a3…am>0,且 a 1 + a 2 + a 3 + … + a m = N a_1+a_2+a_3+…+a_m=N a1+a2+a3+…+am=N的一个有序集合。但是整数的lqp拆分要求的不是拆分总数,相对更加困难一些。
对于每个拆分,lqp定义这个拆分的权值 F a 1 F a 2 … F a m F_{a1}F_{a2}…F_{am} Fa1Fa2…Fam,他想知道对于所有的拆分,他们的权值之和是多少?
简单来说,就是求
∑
∏
i
=
1
m
F
a
i
\sum\prod_{i=1}^m F_{a_i}
∑∏i=1mFai
m
>
0
m>0
m>0
a
1
,
a
2
.
.
.
a
m
>
0
a_1,a_2...a_m>0
a1,a2...am>0
a
1
+
a
2
+
.
.
.
+
a
m
=
N
a_1+a_2+...+a_m=N
a1+a2+...+am=N
由于这个数会十分大,lqp稍稍简化了一下题目,只要输出对于 N N N的整数lqp拆分的权值和 m o d ( 1 0 9 + 7 ) mod (10^9+7) mod(109+7)输出即可。
输入输出格式
输入格式:
输入的第一行包含一个整数 N N N( N ≤ 1 0 6 N \le 10^6 N≤106)。
输出格式:
输出一个整数,为对于 N N N的整数lqp拆分的权值和 m o d ( 1 0 9 + 7 ) mod (10^9+7) mod(109+7)。
输入输出样例
输入样例#1:
3
输出样例#1:
5
说明
F
0
=
0
,
F
1
=
1
,
F
2
=
1
,
F
3
=
2
F_0=0,F_1=1,F_2=1,F_3=2
F0=0,F1=1,F2=1,F3=2。
对于
N
=
3
N=3
N=3,有这样几种lqp拆分:
3
=
1
+
1
+
1
3=1+1+1
3=1+1+1, 权值是
1
∗
1
∗
1
=
1
1*1*1=1
1∗1∗1=1。
3
=
1
+
2
3=1+2
3=1+2,权值是
1
∗
2
=
2
1*2=2
1∗2=2。
3
=
2
+
1
3=2+1
3=2+1,权值是
2
∗
1
=
2
2*1=2
2∗1=2。
所以答案是
1
∗
1
∗
1
+
1
∗
2
+
2
∗
1
=
5
1*1*1+1*2+2*1=5
1∗1∗1+1∗2+2∗1=5。
解题分析
设
F
(
x
)
F(x)
F(x)为斐波那契数列的生成函数,
G
(
x
)
G(x)
G(x)为答案的生成函数, 那么有:
a
n
s
[
i
]
=
[
x
i
]
∑
k
=
1
∞
F
k
(
x
)
=
[
x
i
]
F
(
x
)
(
1
−
F
∞
(
x
)
)
1
−
F
(
x
)
=
[
x
i
]
F
(
x
)
1
−
F
(
x
)
ans[i]=[x^i]\sum_{k=1}^{\infin}F^k(x) \\ =[x^i]\frac{F(x)(1-F^\infin(x))}{1-F(x)} \\ =[x^i]\frac{F(x)}{1-F(x)}
ans[i]=[xi]k=1∑∞Fk(x)=[xi]1−F(x)F(x)(1−F∞(x))=[xi]1−F(x)F(x)
那么现在问题在于如何求出
F
(
x
)
F(x)
F(x)。
可以这样构造:
0 1 0 0 0... (x)
0 0 f(1) f(2) f(3)... (xF(x))
0 0 0 f(1) f(2)... (x^2F(x))
就有 F ( x ) = x + x F ( x ) + x 2 F ( x ) F(x)=x+xF(x)+x^2F(x) F(x)=x+xF(x)+x2F(x)。
解得 F ( x ) = 1 1 − x − x 2 F(x)=\frac{1}{1-x-x^2} F(x)=1−x−x21。
代入得 G ( x ) = x 1 − 2 x − x 2 G(x)=\frac{x}{1-2x-x^2} G(x)=1−2x−x2x
就有 G ( x ) = 2 x G ( x ) + x 2 G ( x ) + x G(x)=2xG(x)+x^2G(x)+x G(x)=2xG(x)+x2G(x)+x
而我们知道, x k x^k xk的意义是将所有数列向右平移 k k k位, 所以就有 d p [ i ] = 2 d p [ i − 1 ] + d p [ i − 2 ] dp[i]=2dp[i-1]+dp[i-2] dp[i]=2dp[i−1]+dp[i−2]。
x x x的意义是 d p [ 1 ] = 1 dp[1]=1 dp[1]=1。
代码如下:
#include <cstdio>
#define R register
#define MOD 1000000007
int dp[1005000];
int main(void)
{
int n;
scanf("%d", &n); dp[1] = 1;
for (R int i = 2; i <= n; ++i) dp[i] = (dp[i - 1] * 2 % MOD + dp[i - 2]) % MOD;
printf("%d", dp[n]);
}