杭电OJ [1005](http://acm.hdu.edu.cn/showproblem.php?pid=1005): #####Problem Description > A number sequence is defined as follows: f(1) = 1, f(2) = 1, f(n) = (A * f(n - 1) + B * f(n - 2)) mod 7. Given A, B, and n, you are to calculate the value of f(n). #####Input > The input consists of multiple test cases. Each test case contains 3 integers A, B and n on a single line (1 <= A, B <= 1000, 1 <= n <= 100,000,000). Three zeros signal the end of input and this test case is not to be processed. #####Output > For each test case, print the value of f(n) on a single line. #####Sample Input > 1 1 3 1 2 10 0 0 0 #####Sample Output > 2 5 题目对于那些ACM选手来讲,确定不是什么大问题,不过对于咱们这种只能刷刷水题的人来讲,仍是有点困难的。 我看到题目以后的第一反应是对于每个特定的A和B,每次的计算结果都存到数组里,这样不用每一个输入都从新计算,能够节省必定量的时间,当时以为这个想法已经不错了,可是仍是TLE了。最后上网找了答案,发现不少答案里都提到f(n)的值实际上是循环的,并且最大的循环长度不会超过49(即起码在n=49以前,f(n)的值就开始循环了,f(k)=f(1),f(k+1)=f(2),f(k+2)=f(3),...,k<=49)。可是网上不少文章并无指出怎么才能发现,或者说推导出这个规律。最后我花了点时间,本身推了下才终于知道了发现规律的方法。 首先,观察到递推式里有mod 7,就知道f(n)的全部值都在[0, 6]之间,有7个取值。 而后,观察递推式,f(n) = (A * f(n - 1) + B * f(n - 2)) mod 7,A和B是定常数,而f(n - 1)和f(n - 2)的取值都分别有7种可能,也就是说f(n)的取值最多有49种组合(这49种组合中,和可能相同,可是表明的意义不一样,例如1+4和2+3是不一样的,2+3和3+2也是不一样的)。当n > 51时(由于这种组合是从n=3开始算的),f(n)的取值组合必然是以前出现过了的,也就是必存在 ``` f(n) = f(k), f(n - 1) + f(n - 2) = f(k - 1) + f(k - 2) f(n - 1) = f(k - 1) f(n - 2) = f(k - 2) ``` 可是其实咱们能够知道,f(n) = 0 + 0,这种状况是不可能的(由于这样话很容易证实对于全部的n,f(n)都是为0),因此其实最多只有48种状况,即从n = 50开始,组合就必然出现重复了。 其次,咱们知道了f(n)和f(k)的取值组合彻底相同后,只要证实f(n + 1) = f(k + 1)的便可证实f(n)的取值在n > 49后必然存在循环。 ``` f(n + 1) = f(n) + f(n - 1) f(n) = f(k) f(n - 1) = f(k - 1) ``` 由以上条件可知,f(n + 1) = f(k + 1),同理能够推导出f(n + 2) = f(k + 2),...,等等。并最终证实f(n)的值是循环的。 最后,咱们已经证实了f(n)是存在循环的,最后要证实的是f(n)是整循环的,也就是说循环起始点应该是f(n) = f(n - 1) + f(n - 2) = f(1) + f(2)。先假设f(n + 2) = f(5) = f(n + 1) + f(n) = f(4) + f(3),n是循环开始点,由假设能够推导出f(n + 1) = f(n) + f(n - 1) = f(4) + f(3),f(n - 1)和f(3)以前存在两种可能: 1. f(n - 1) = f(3) 2. |f(n - 1) - f(3)| = 7 由f(n)的取值范围[0, 6]知,第二种状况是不可能的,因此f(n - 1) = f(3),因此n - 1是循环开始点,依次能够类推n - 2是循环开始点,...,直到f(n - k) = f(3) = f(n - k - 1) + f(n - k - 2) = f(2) + f(1),n - k - 2是循环开始点。因此由证实可知,若是n是循环开始点,则f(n) = f(1),f(n + 1) = f(2),...。 知道了f(n)的值是循环的以后,这道题目就很容易作了,只要求出循环开始点就好了,即f(i - 1) = 1, f(i) = 1。 具体实现代码以下:
#include <iostream> using namespace std; int f[50] = {0, 1, 1}; int a, b, n; int main() { while (cin >> a >> b >> n) { if (a == 0 && b == 0 && n == 0) { break; } if (n > 2) { int i; for (i = 3; i <= 49; i++) { f[i] = (a * f[i - 1] + b * f[i - 2]) % 7; if (f[i] == f[2] && f[i - 1] == f[1]) { break; } } i -= 2; n = n % i == 0 ? i : n % i; } cout << f[n] << endl; } }