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

题目描述

给定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 <stdio.h>
#include <stdlib.h>
 
void MetrixMul(int p1[2][2],int p2[2][2])
{
    int t[2][2];
 
    t[0][0]=(p1[0][0]*p2[0][0]%10000+p1[0][1]*p2[1][0]%10000)%10000;
    t[0][1]=(p1[0][0]*p2[0][1]%10000+p1[0][1]*p2[1][1]%10000)%10000;
    t[1][0]=(p1[1][0]*p2[0][0]%10000+p1[1][1]*p2[1][0]%10000)%10000;
    t[1][1]=(p1[1][0]*p2[0][1]%10000+p1[1][1]*p2[1][1]%10000)%10000;
 
    p1[0][0]=t[0][0];
    p1[0][1]=t[0][1];
    p1[1][0]=t[1][0];
    p1[1][1]=t[1][1];
 
}
 
void MetrixPow(int p[2][2],int n)
{
    int t[2][2];
 
    t[0][0]=p[0][0];
    t[0][1]=p[0][1];
    t[1][0]=p[1][0];
    t[1][1]=p[1][1];
 
    if(n==1)
        return;
    
    else if(n%2)
    {
        MetrixPow(p,n-1);
        MetrixMul(p,t);
    }
    else
    {
        MetrixPow(p,n/2);
        MetrixMul(p,p);
    }
}
 
int main()
{
    int a0,a1,p,q,k;
    int pa[2][2];
 
    while(scanf("%d%d%d%d%d",&a0,&a1,&p,&q,&k)!=EOF)
    {
        if(k==0)
        {
            printf("%d\n",a0%10000);
            continue;
        }
        if(k==1)
        {
            printf("%d\n",a1%10000);
            continue;
        }
 
        pa[0][0]=p%10000;
        pa[0][1]=q%10000;
        pa[1][0]=1;
        pa[1][1]=0;
 
        MetrixPow(pa,k-1);
 
        printf("%d\n",(pa[0][0]*a1+pa[0][1]*a0)%10000);
    }
    return 0;
}