机试题---递推数列

题目：

题目描述

给定a0,a1,以及an=p*a(n-1) + q*a(n-2)中的p,q。这里n >= 2。 求第k个数对10000的模。

输入描述:

输入包括5个整数：a0、a1、p、q、k。

输出描述:

第k个数a(k)对10000的模。

示例1

输入

复制

20 1 1 14 5

输出

复制

8359

代码：

#include<iostream>
#include<vector>
using namespace std;

int main() {

	int a0, a1, p, q, k;

	while (cin >> a0 >> a1 >> p >> q >> k) {

		vector<long long> a;
		a.push_back(a0);
		a.push_back(a1);

		for (int i = 2; i <= k; i++) {
			int num = (p*a[i - 1] + q*a[i - 2])%10000;
			a.push_back(num);
		}

		cout << a[k] << endl;
	}
	return 0;
}

突破点：

(a+b) mod N=((a mod N)+(b mod N))mod N