【循环节】数列 sequence

数列(sequence)

Sequence.cpp/c/pas/1s/256M

【问题描述】

一个数列定义如下：f(1) =1, f(2) = 1, f(n) = (A * f(n - 1) + B * f(n - 2)) mod 7。给

定A, B 和n 的值，要求计算f(n)的值。

【输入文件】

输入文件sequence.in仅一行包含3 个整数A，B 和n，其中(1 ≤ A, B ≤ 1000,

1 ≤ n ≤ 100,000,000)。

【输出文件】

输出文件sequence.out仅一行，一个整数，即f(n)的值。

【输入样例】

1 1 3

【输出样例】

2

【数据规模】

20%的数据，n ≤1,000。

40%的数据，n ≤100,000。

100%的数据，n ≤ 100,000,000。

题目描述很简单，就一个简单的递推式，但是看看范围，根本存不下！

先说说非完美算法，虽然存不下，但是从题目得知当前的 f [ i ] 只和前面的 f [ i - 1 ] 和 f [ i - 2 ] 有关，所以我们可以用 x y z 每次模3来实现循环滚动，但是仍要超时！

一个小技巧就是把 int 改成 short ，那么不但可以存下所有数据，还可以AC！！（不过够悬的~~~）

/*
C++ Code
http://blog.csdn.net/jiangzh7
By Jiangzh
*/
#include<cstdio>
#include<map>
using namespace std;

int a,b,n;
short f[100000000];

int main()
{
	freopen("sequence.in","r",stdin);
	freopen("sequence.out","w",stdout);
	scanf("%d%d%d",&a,&b,&n);
	f[1]=f[2]=1;
	for(int i=3;i<=n;i++) f[i]=(a*f[i-1]+b*f[i-2])%7;
	printf("%d\n",f[n]);
	return 0;
}

在说说正解~

这种题是肯定有循环节的！至于证明，我们先看一维的递推式 f [ i ] = A * f [ i - 1 ]  ，假设我们在后面找到一个 f [ x ] 和前面的 f [ t ] 相等，那么我们将 f [ x ] 和 f [ t ]  带入前面的递推式是肯定会得到一样的答案的，所以这就进入了所谓的循环节

然后再看看这里的 f [ i ] = A * f [ i -1 ] + B * f[ i - 2 ]，当前值是由前两个决定的，所以我们只要发现有两个连续的数再次出现的话，后面就一定可以推到一样的 f [ i ] ！所以就可以找到循环节！不过值得注意的是，循环节可能不是从第一个开始的！

C++ Code

/*
C++ Code
http://blog.csdn.net/jiangzh7
By Jiangzh
*/
#include<cstdio>
#include<cstdlib>

typedef long long LL;
int A,B,n;
int f[100];//will use it at most 49 times
int Ts,Te,T;//cycle
int h[10][10];

void read()
{
	freopen("sequence.in","r",stdin);
	freopen("sequence.out","w",stdout);
	scanf("%d%d%d",&A,&B,&n);
}

void find_T()//find cycle
{
	f[1]=f[2]=1;
	h[1][1]=0;
	for(int i=3;i<=3+49;i++)
	{
		f[i]=(A*f[i-1]+B*f[i-2])%7;
		if(h[f[i-1]][f[i]])
		{
			Ts=h[f[i-1]][f[i]];
			Te=i-2; T=Te-Ts;
			break;
		}
		else h[f[i-1]][f[i]]=i-2;
	}
}

void work()
{
	find_T();
	printf("%d",f[Ts+(n-Ts)%T]);
}

int main()
{
	read();
	work();
	return 0;
}