码迷,mamicode.com
首页 > 其他好文 > 详细

洛谷1349 广义斐波那契数列 【矩阵乘法】

时间:2015-10-15 14:15:31      阅读:215      评论:0      收藏:0      [点我收藏+]

标签:

洛谷1349 广义斐波那契数列

题目描述

广义的斐波那契数列是指形如an=p*an-1+q*an-2的数列。今给定数列的两系数p和q,以及数列的最前两项a1和a2,另给出两个整数n和m,试求数列的第n项an除以m的余数。

输入输出格式

输入格式:

输入包含一行6个整数。依次是p,q,a1,a2,n,m,其中在p,q,a1,a2整数范围内,n和m在长整数范围内。

输出格式:

输出包含一行一个整数,即an除以m的余数。

输入输出样例

输入样例#1:

1 1 1 1 10 7

输出样例#1:

6

说明

数列第10项是55,除以7的余数为6。

 

 

【思路】

  矩阵乘法。

  n最大为maxlongint所以枚举肯定会有TLE。

  这里用到了矩阵乘法。大体思想:因为每次的求解规则相同,所以可以构造一个转移矩阵,使得每个点与该矩阵相乘即可直接得出最终该点的位置。

  开始矩阵B:

      a2 a1

构造矩阵如下:

      P 1

      Q 0

  (一次转换后为 p*a2+q*a1,a2 )

  因为要进行n-2次相同的转换所以用快速幂求解转换n-2次后的A’ 。

  ans=B*A‘

 

  注意: 矩阵乘法的顺序不能更改,否则可能不满足乘法要求。

         LL 与 LL的乘法可能会爆精度,可以转化为小数相加的形式。

 

代码思想源于洛谷题解。

【代码】

 

 

 1 #include<iostream>
 2 #include<cstring>
 3 using namespace std;
 4 
 5 typedef long long LL;
 6 const int delen=23,delta=1<<23,deltb=delta-1;
 7 LL n,m,p,q,a1,a2;
 8 
 9 LL Add(LL x,LL y) {
10     x+=y;
11     if(x>=m) x-=m;
12     return x;
13 }
14 
15 LL multi(LL x,LL y) {  //对于long long型的 乘法+模 防止越界 
16     LL ans=0;
17     while(y) {
18         if(y&deltb) ans=Add(ans,x*(y&deltb)%m);
19         x=x*delta%m;
20         y>>=delen;
21     }
22     return ans;
23 }
24 
25 struct Matrix{
26     int r,c;
27     LL N[5][5];
28     Matrix operator *(const Matrix& rhs) const{
29         Matrix tmp={};
30         tmp.r=r, tmp.c=rhs.c;
31         for(int i=0;i<tmp.r;i++)
32           for(int j=0;j<tmp.c;j++) {
33                 for(int k=0;k<c;k++)  tmp.N[i][j] += multi(N[i][k],rhs.N[k][j]);
34                 tmp.N[i][j] %= m;
35           }
36         return tmp;
37     }
38     
39     Matrix pow(LL p) const{
40         Matrix ans={} , tmp=*this;
41         ans.r=ans.c=r;
42         for(int i=0;i<r;i++) ans.N[i][i]=1;
43         while(p) {
44             if(p&1) ans=ans*tmp;
45             tmp=tmp*tmp;
46             p>>=1; 
47         }
48         return ans;
49     }
50 };
51 
52 int main() {
53     cin>>p>>q>>a1>>a2>>n>>m;
54     Matrix A,B;
55     A.r=A.c=2;
56     A.N[0][0]=p , A.N[1][0]=q , A.N[0][1]=1 , A.N[1][1]=0; //0
57     A=A.pow(n-2);
58     
59     B.r=1 , B.c=2;
60     B.N[0][0]=a2,B.N[0][1]=a1;
61     
62     B=B*A;    //B*A
63     
64     cout<<B.N[0][0]%m;
65     return 0;
66 }

 

洛谷1349 广义斐波那契数列 【矩阵乘法】

标签:

原文地址:http://www.cnblogs.com/lidaxin/p/4882141.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!