这道题又一次更新了我的世界观与人生观Orz……最开始我是设计了一个O(n)的递推算法,本以为可以轻松AC没想到居然TLE了……然后搜了一下题解,才发现这道题要用矩阵的思想去做。
通过对题目的分析,我们可以得到矩阵递推公式如下:
将公式右边推至a1,a0即可得:
然后这个题的关键就转化成了求[p q;1 0]的(k-1)次幂的问题。
接下来求矩阵的幂可以用快速幂运算来解决,这样就将O(n)的算法化简为了O(logn)的复杂度,如此一来就可以AC了。
#include <stdio.h> #include<stdlib.h> #include<string.h> #define NUM 10000 typedef long long ll; typedef struct { ll num[2][2]; }Matrix; ll a1,a0,p,q,k; Matrix mul(Matrix a1,Matrix a2) { Matrix solution; int i,j; for(i=0;i<=1;i++) for(j=0;j<=1;j++) { solution.num[i][j]=(a1.num[i][0]*a2.num[0][j]+a1.num[i][1]*a2.num[1][j])%NUM; } return solution; } Matrix a,res; void calculate(ll n) { res.num[0][0]=1; res.num[1][1]=1; res.num[1][0]=0; res.num[0][1]=0; while(n) { if(n&1) res=mul(res,a); n>>=1; a=mul(a,a); } ll n1=(res.num[0][0]*a1+res.num[0][1]*a0)%NUM; //ll n2=(res.num[1][0]*a1+res.num[1][1]*a0)%NUM; printf("%lld\n",n1); } int main() { ll i,j; while(scanf("%lld%lld%lld%lld%lld",&a0,&a1,&p,&q,&k)!=EOF) { if(k==0) { printf("%lld\n",a0%NUM); } else if(k==1) { printf("%lld\n",a1%NUM); } else { a.num[0][0]=p; a.num[0][1]=q; a.num[1][0]=1; a.num[1][1]=0; calculate(k-1); } } return 0; }
关于矩阵的快速幂运算,可以参考这篇博文:http://www.cnblogs.com/yan-boy/archive/2012/11/29/2795294.html