nyoj 301 递推求值【Java】

时间限制：1000 ms | 内存限制：65535 KB
难度：4

描述

给你一个递推公式：

f(x)=af(x-2)+bf(x-1)+c

并给你f(1),f(2)的值，请求出f(n)的值，由于f(n)的值可能过大，求出f(n)对1000007取模后的值。

注意:-1对3取模后等于2

输入
第一行是一个整数T,表示测试数据的组数(T<=10000)
随后每行有六个整数，分别表示f(1),f(2),a,b,c,n的值。
其中0<=f(1),f(2)<100,-100<=a,b,c<=100,1<=n<=100000000 (10^9)
输出
输出f(n)对1000007取模后的值

样例输入

2
1 1 1 1 0 5
1 1 -1 -10 -100 3

样例输出

5
999896

代码

import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		Scanner cin = new Scanner(System.in);

		int m = cin.nextInt();
		while (m > 0) {
			int f1 = cin.nextInt();
			int f2 = cin.nextInt();
			int a = cin.nextInt();
			int b = cin.nextInt();
			int c = cin.nextInt();
			int n = cin.nextInt();
			int[] arr = new int[n];
			if (n == 1) {
				arr[0] = f1;
				System.out.println(arr[0]);
				continue;
			}
			arr[0] = f1;
			arr[1] = f2;
			for (int i = 2; i < n; i++) {
				arr[i] = a * arr[i - 2] + b * arr[i - 1] + c;
				if (arr[i] < 0)
					arr[i] = 1000007 + arr[i];
				arr[i] = arr[i] % 1000007;

			}
			System.out.println(arr[n - 1]);
			m--;
		}

		cin.close();
	}

}