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

矩阵快速幂

时间:2018-05-14 22:06:00      阅读:236      评论:0      收藏:0      [点我收藏+]

标签:include   输入   pre   des   close   opened   amp   记录   接下来   

写在前面

 

快速幂思想:

降低时间复杂度,使用尽量少的相乘次数。

 

例如:q^w化为(q*q)^(w/2) 。注意:当w为奇数的时候,我们要把(q*q)^(w/2)乘以q,理由是符号/”是整除。所以在快速幂中乘积的底数是q*q,不是q了。

 

展示实例

问题描述

 

 给定一个N阶矩阵A,输出A的M次幂(M是非负整数)

  例如:

  A =

  1 2

  3 4

  A的2次幂

  7 10

  15 22

输入格式

      第一行是一个正整数N、M(1<=N<=30, 0<=M<=5),表示矩阵A的阶数和要求的幂数。接下来N行,每行N个绝对值不超过10的非负整数,描述矩阵A的值。

输出格式

  输出共N行,每行N个整数,表示A的M次幂所对应的矩阵。相邻的数之间用一个空格隔开。

样例输入

2 2
1 2
3 4

样例输出

7 10
15 22

 

思路解析

用矩阵快速幂优化时间复杂度。

代码详解

 1 #include<stdio.h>
 2 #include<string.h>
 3 using namespace std;
 4 struct M
 5 {
 6     int num[35][35];
 7     M()
 8     {
 9         memset(num,0,sizeof(num));
10     }
11 } a,e;
12 int m;
13 M chengji(M a,M b)//两个矩阵乘积结果得到一个矩阵
14 {
15     M q;
16     for(int i=0; i<m; i++)
17     {
18         for(int j=0; j<m; j++)
19         {
20             for(int k=0; k<m; k++)
21             {
22                 q.num[i][j]+=(a.num[i][k]*b.num[k][j]);
23             }
24         }
25     }
26     return q;
27 }
28 M mulPower(M c,int m) //矩阵快速幂
29 {
30     M b=c,ans=e;
31     while(m)
32     {
33         if(m&1)
34         {
35             ans=chengji(ans,b);
36         }
37         b=chengji(b,b);
38         m>>=1;
39     }
40     return ans;
41 }
42 int main()
43 {
44     int n;
45     scanf("%d %d",&n,&m);
46     for(int i=0; i<m; i++)
47     {
48         e.num[i][i]=1;
49         for(int j=0; j<m; j++)
50         {
51             scanf("%d",&a.num[i][j]);
52         }
53     }
54     M x = mulPower(a,n);
55     for(int i=0; i<m; i++)
56     {
57         for(int j=0; j<m; j++)
58         {
59             printf("%d ",x.num[i][j]);
60 
61         }
62         printf("\n");
63     }
64     return 0;
65 }

 

总结提升

 

 

矩阵快速幂的实例:

斐波那契数列。它的递推公式为:F[n] = F[n-1] + F[n-2] 。 可以用矩阵表示:

技术分享图片

 

矩阵快速幂算法实现:

技术分享图片
 1 struct Matrix
 2 {
 3     int a[maxn][maxn];
 4 }ans,ju;
 5 
 6 //计算矩阵乘法的函数,参数是两个矩阵和阶数n
 7 Matrix Mul(Matrix A,Matrix B,int n)
 8 {
 9     Matrix tmp;
10     int i,j,k;
11     //初始化tmp矩阵
12     for(i=0;i<n;i++)
13     {
14         for(j=0;j<n;j++)
15         {
16             tmp.a[i][j]=0;
17         }
18     }
19     //两个矩阵乘积的结果
20     for(i=0;i<n;i++)
21     {
22         for(j=0;j<n;j++)
23         {
24             for(k=0;k<n;k++)
25             {
26                 tmp.a[i][j]=A.a[i][k]+B.a[k][j];
27             }
28         }
29     }
30     return tmp;
31 }
32 //快速幂算法,求矩阵ju的m次幂
33 void QuickPower(int m,int n)
34 {
35     //矩阵ans初始为单位矩阵,就是主对角线为1,其他为0
36     int i,j;
37     for(i=0;i<n;i++)
38     {
39         for(j=0;j<n;j++)
40         {
41             if(i==j)ans.a[i][j]=1;
42             else
43                 ans.a[i][j]=0;
44         }
45     }
46     //如果是0,只需要直接把底数a扩大,也就是a*a...,不需要把a记录到答案ans
47     while(m)
48     {
49         if(m&1)
50             ans=Mul(ans,a);
51         a=Mul(a,a);
52         a=a>>1;
53     }
54 }
View Code

 

技术分享图片

技术分享图片

 

 

 

 

 

 技术分享图片

 

矩阵快速幂

标签:include   输入   pre   des   close   opened   amp   记录   接下来   

原文地址:https://www.cnblogs.com/weixq351/p/9038298.html

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