CF450B-Jzzhu and Sequences

题目链接

题目描述

Jzzhu发明了一种序列，它们有以下性质：

f(1)=x; f(2)=y; (对任意i, i>=2) , f(i)=f(i-1)+f(i+1)

给出 x,y ,现请你计算出 f(n) mod 1000000007 (10^9+7) 的值。

输入

第一行包含2个整数x和y，（|x|, |y|<= 10^{9）。第二行包含1个整数n，（1&lt;=n&lt;=2*10}9）。

输出

一行。输出一个整数** f(n)**模1000000007（10^9+7)的值

解题思路

把题目里的公式变换一下，有：$F_i=F_{i-1}-F_{i-2}$，
如果我们直接通过递推枚举的话，一定会超时，把上面的公式展开，我们可以得到：

$F_i=F_{i-1}-F_{i-2}$
$F_i=(F_{i-2}-F_{i-3})-F_{i-2}$
$F_i=-F_{i-3}$

继续展开$F_{i-3}$，有：

$F_i=-F_{i-3}$
$F_i=-(F_{i-4}-F_{i-5})$
$F_i=F_{i-6}$

通过上面的推导我们可以知道，每3个数一个轮回，偶数次的轮回需要乘以-1.

如果有心的话，使用检举法写一个程序把所有项输出，也很容易发现规律。这样做还有一个好处就是，根据暴力程序的结果，来验证你写的优化程序是否正确。

枚举的代码如下：

#include <iostream>
#include <cmath>
using namespace std;
int f[10086];
int n;
int main(){
	cin>>f[1]>>f[2]>>n;
	for(int i=3;i<=n;i++){
		f[i]=f[i-1]-f[i-2];
	}
	for(int i=1;i<=n;i++)
		cout<<f[i]<<endl;
	return 0;
}

运行结果如下：

参考代码

#include <iostream>
#include <cmath>
using namespace std;
const int MOD=1E9+7;
long long f1,f2,f3,ans;

int main(){
	int cnt,n,flag=1;
	cin>>f1>>f2>>n;
	f1=(f1+MOD)%MOD;
	f3 = f2-f1;
	switch( n%3){
		case 1: ans=f1;break;
		case 2: ans=f2;break;
		case 0: ans=f3;break;
	}
	cnt = ceil( n/3.0 );
	if ( cnt%2 == 0) 
		flag=-1;
	ans = ( flag*ans%MOD+MOD)%MOD;
	cout<<ans<<endl;
}