考研机试 递推数列

递推数列

给定a0,a1,以及an=pa(n-1) + qa(n-2)中的p,q。这里n >= 2。 求第k个数对10000的模。
输入描述：
输入包括5个整数：a0、a1、p、q、k。
输出描述：
第k个数a(k)对10000的模。

思路使用vector数组存放从0到k的数据 因为最后求的是对10000的模 为了保证数据不超过int的范围每个位置实际上可以存放实际的值%10000 最后下标为k的位置存放的就是a(k)对10000的模

#include <cstdio>
#include <vector>
using namespace std;
int main(){
    int a0,a1,p,q,k;
    scanf("%d%d%d%d%d",&a0,&a1,&p,&q,&k);
    vector<int> vec1(10001);
    vec1[0]=a0;vec1[1]=a1;
    if(k==0||k==1){
        printf("%d\n",vec1[k]%10000);
    }
    else{
        for(int i=2;i<=k;++i){
        vec1[i]=(p*vec1[i-1]+vec1[i-2]*q)%10000;
        }
        printf("%d\n",vec1[k]);
    }
}