java 杭电acm1005_【2020杭电多校round1 1005】HDU6755 Fibonacci Sum（二项式定理）

最新推荐文章于 2021-04-11 14:52:43 发布

余女士

最新推荐文章于 2021-04-11 14:52:43 发布

阅读量87

点赞数

文章标签： java 杭电acm1005

本文链接：https://blog.csdn.net/weixin_28020031/article/details/115042287

版权

该博客介绍了如何利用斐波那契数列的通项公式和二项式定理解决杭电ACM1005问题，涉及大数模运算和等比数列求和，给出O(K log N)复杂度的解决方案。

摘要由CSDN通过智能技术生成

题目大意

定义斐波那契数列：

\[F_0=0,F_1=1\\ F_n=F_{n-1}+F_{n-2}\ (n>1) \]

给定整数\(N,C,K\)。请计算下式的值：

\[(F_{0})^K+(F_{C})^K+(F_{2C})^K+\dots +(F_{NC})^{K} \]

因为答案太大，请输出其对\(1000000009\) (\(10^9+9\)) 取模的值。

数据范围：\(1\leq N,C\leq 10^{18}\)，\(1\leq K\leq 10^5\)。

本题题解

首先需要了解斐波那契数列的通项公式：

\[F_n=\frac{1}{\sqrt{5}}\left(\left(\frac{1+\sqrt{5}}{2}\right)^{n}-\left(\frac{1-\sqrt{5}}{2}\right)^{n}\right) \]

通过暴力，可以找到\(\bmod10^9+9\)意义下\(\sqrt{5}\)的值是\(383008016\)。

const int MOD=1e9+9;

for(int i=0;i

if((long long)i*i%MOD == 5){

cout<

break;

}

考虑求答案。先把\(\frac{1}{\sqrt{5}}\)提取出来，即：

\[\begin{align} \text{answer}&=\sum_{i=0}^{N}(F_{iC})^{K}\\ &=\sum_{i=0}^{N}\left(\frac{1}{\sqrt{5}}\left(\left(\frac{1+\sqrt{5}}{2}\right)^{iC}-\left(\frac{1-\sqrt{5}}{2}\right)^{iC}\right)\right)^K\\ &=\left(\frac{1}{\sqrt{5}}\right)^K\cdot \sum_{i=0}^{N}\left(\left(\frac{1+\sqrt{5}}{2}\right)^{iC}-\left(\frac{1-\sqrt{5}}{2}\right)^{iC}\right)^{K} \end{align} \]

设\(A=\left(\frac{1+\sqrt{5}}{2}\right)^C\), \(B=\left(\frac{1-\sqrt{5}}{2}\right)^C\)。则：

\[\text{answer}=\left(\frac{1}{\sqrt{5}}\right)^K\sum_{i=0}^{N}(A^i-B^i)^K \]

问题转化为求：

\[\sum_{i=0}^{N}(A^i-B^i)^K \]

把后面的东西用二项式定理展开，得到：

\[=\sum_{i=0}^{N}\sum_{j=0}^{K}{K\choose j}A^{ij}(-B^i)^{K-j} \]

交换和式，得：

\[=\sum_{j=0}^{K}{K\choose j}(-1)^{K-j}\sum_{i=0}^{N}(A^j)^i(B^{K-j})^i \]

后面的\(\sum_{i=0}^{N}(A^j)^i(B^{K-j})^i\)，是一个等比数列求和，可以\(O(\log N)\)求出来。(即\(\sum_{i=0}^{n}x^i=\frac{x^{n+1}-1}{x-1}\)，\(x\neq 1\))。复杂度里的\(\log\)来自求\(x^{n+1}\)时使用的快速幂。

于是总时间复杂度就是\(O(K\log N)\)的了。

参考代码：

#include

using namespace std;

typedef long long ll;

const int MOD=1e9+9;

const int MAXK=1e5;

int pow_mod(int x,int i){

int y=1;

while(i){

if(i&1)y=(ll)y*x%MOD;

x=(ll)x*x%MOD;