3392. 递推数列(清华大学考研机试题)

给定 $a_0$,$a_1$，以及 $a_n$=$p$×$a_{n-1}$+$q$×$a_{n-2}$ 中的$p$,$q$。

这里 n≥2。

求第 k 个数 $a_k$ 对 10000 的模。

输入格式
输入包括 5 个整数：$a_0$、$a_1$、$p$、$q$、$k$。

输出格式
第 $k$个数 $a_k$ 对 10000 的模。

数据范围
1≤$a_0$,$a_1$,$p$,$q$,$k$≤10000

输入样例：
20 1 1 14 5

输出样例：
8359

My Answer Code:

/*
	Author:Albert Tesla Wizard
	Time:2021/9/2 10:59
*/
#include<bits/stdc++.h>
using namespace std;
int p,q,k,a0,a1,f;
void solve()
{
    if(k==0)f=a0%10000;
    else if(k==1)f=a1%10000;
    else
    {
        for(int i=0;i<k-1;i++)
        {
            f=(p*a1+q*a0)%10000;
            a0=a1;
            a1=f;
        }
    }
    cout<<f<<'\n';
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin>>a0>>a1>>p>>q>>k;
    solve();
    return 0;
}