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

Gauss Fibonacci HDU - 1588 等比矩阵列求和

时间:2017-10-02 09:31:09      阅读:224      评论:0      收藏:0      [点我收藏+]

标签:print   mem   stdout   code   operator   def   sizeof   memset   desktop   

二分求和
或者矩阵套矩阵
|A E| ^n = |A^n E+...+A^(n-1)|
|0 E| |0 E |
https://cn.vjudge.net/solution/9721123
这个代码我却不懂了= =



#include <stdio.h> #include <cstring> #include <algorithm> #include <queue> #include <math.h> #include <iostream> using namespace std; #define LL long long const int maxn=(int)1e5+5; LL MOD=8000; int n; struct mat{ LL a[2][2]; mat(){ memset(a,0,sizeof a); } mat operator *(const mat &q){ mat t; for(int i=0;i<n;i++) for(int j=0;j<n;j++ ){ LL w=0; for(int k=0;k<n;k++)w=(w+a[i][k]*q.a[k][j])%MOD; t.a[i][j]=w; } return t; } mat operator +(const mat& q){ mat t;memset(t.a,0,sizeof t.a); for(int i=0;i<n;i++) for(int j=0;j<n;j++){ t.a[i][j]=(a[i][j]+q.a[i][j])%MOD; } return t; } void initE(){ a[1][0]=a[0][1]=0;a[1][1]=a[0][0]=1; } void initFib(){ for(int i=0;i<n;i++)for(int j=0;j<n;j++)a[i][j]=1;a[1][1]=0; } void in(){ for(int i=0;i<n;i++) for(int j=0;j<n;j++)scanf("%lld",&a[i][j]); } void print(){ for(int i=0;i<n;i++){ for(int j=0;j<n;j++)printf("%lld ",a[i][j]);printf("\n"); } } }; LL k,b,N; mat quickmatpower(mat a ,LL k){ mat res;res.initE(); while(k>0){ if(k&1)res=res*a; a=a*a; k>>=1LL; } return res; } LL quickpower(LL a,LL k){ LL res=1; while(k>0){ if(k&1)res=res*a; a=a*a; k>>=1LL; } return res; } mat calsum(mat a,LL k){ mat E;E.initE(); if(k==1) return a; if(k%2==0){ return calsum(a,k/2)*(E+quickmatpower(a,k/2)); } return calsum(a,k/2)*(E+quickmatpower(a,k/2))+quickmatpower(a,k); } LL work(){ mat f;f.initFib();mat E;E.initE(); return (quickmatpower(f,b)*((E+calsum(quickmatpower(f,k),N-1)))).a[0][1]; } int main() { #ifdef shuaishuai freopen("C:\\Users\\hasee\\Desktop\\a.txt","r",stdin); // freopen("C:\\Users\\hasee\\Desktop\\b.txt","w",stdout); #endif n=2; while(~scanf("%lld%lld%lld%lld",&k,&b,&N,&MOD)){ printf("%lld\n",work()); } return 0; }

 

Gauss Fibonacci HDU - 1588 等比矩阵列求和

标签:print   mem   stdout   code   operator   def   sizeof   memset   desktop   

原文地址:http://www.cnblogs.com/MeowMeowMeow/p/7618433.html

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