[Luogu P4451] [BZOJ 2173] [国家集训队]整数的lqp拆分

洛谷传送门
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,a3am>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=Fn1+Fn2(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,a3am>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} Fa1Fa2Fam,他想知道对于所有的拆分,他们的权值之和是多少?

简单来说,就是求
∑ ∏ 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 N106)。

输出格式:

输出一个整数,为对于 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 111=1
3 = 1 + 2 3=1+2 3=1+2,权值是 1 ∗ 2 = 2 1*2=2 12=2
3 = 2 + 1 3=2+1 3=2+1,权值是 2 ∗ 1 = 2 2*1=2 21=2
所以答案是 1 ∗ 1 ∗ 1 + 1 ∗ 2 + 2 ∗ 1 = 5 1*1*1+1*2+2*1=5 111+12+21=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=1Fk(x)=[xi]1F(x)F(x)(1F(x))=[xi]1F(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)=1xx21

代入得 G ( x ) = x 1 − 2 x − x 2 G(x)=\frac{x}{1-2x-x^2} G(x)=12xx2x

就有 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[i1]+dp[i2]

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]);
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值