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

【矩阵加速】 矩阵 快速幂

时间:2014-10-09 22:42:21      阅读:157      评论:0      收藏:0      [点我收藏+]

标签:style   blog   color   io   os   for   sp   div   问题   

矩阵的快速幂是用来高效地计算矩阵的高次方的。将朴素的o(n)的时间复杂度,降到log(n)。

这里先对原理(主要运用了矩阵乘法的结合律)做下简单形象的介绍:

一般一个矩阵的n次方,我们会通过连乘n-1次来得到它的n次幂。

但做下简单的改进就能减少连乘的次数,方法如下:

把n个矩阵进行两两分组,比如:A*A*A*A*A*A => (A*A)*(A*A)*(A*A)

这样变的好处是,你只需要计算一次A*A,然后将结果(A*A)连乘自己两次就能得到A^6,即(A*A)^3=A^6。算一下发现这次一共乘了3次,少于原来的5次。

其实大家还可以取A^3作为一个基本单位。原理都一样:利用矩阵乘法的结合律,来减少重复计算的次数。

回头看看矩阵的快速幂问题,我们是不是也能把它离散化呢?比如A^19 => (A^16)*(A^2)*(A^1),显然采取这样的方式计算时因子数将是log(n)级别的(原来的因子数是n),不仅这样,因子间也是存在某种联系的,比如A^4能通过(A^2)*(A^2)得到,A^8又能通过(A^4)*(A^4)得到,这点也充分利用了现有的结果作为有利条件。下面举个例子进行说明:

现在要求A^156,而156(10)=10011100(2)

也就有A^156=>(A^4)*(A^8)*(A^16)*(A^128) 考虑到因子间的联系,我们从二进制10011100中的最右端开始计算到最左端。细节就说到这,下面给核心代码:

1 while(N)
2 {
3                 if(N&1)
4                        res=res*A;
5                 n>>=1;
6                 A=A*A;
7  }

里面的乘号,是矩阵乘的运算,res是结果矩阵。

第3行代码每进行一次,二进制数就少了最后面的一个1。二进制数有多少个1就第3行代码就执行多少次。

 

poj 3070 矩阵加速求斐波那契数列代码:

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <memory.h>

using namespace std;
int n = 2, m;
int s[2][2]={{1, 1}, {1, 0}};
struct matrix
{
    int r, c;
    int mat[3][3];
} res, origin;
void init()
{
    memset(res.mat, 0, sizeof(res.mat));
    res.r = n;
    res.c = n;
    for(int i = 1; i <= n; i++)
    {
        res.mat[i][i] = 1;
    }
    origin.c = n;
    origin.r = n;
    for(int i = 1; i <= n; i++)
    {
        for(int j = 1; j <= n; j++)
        origin.mat[i][j] = s[i - 1][j - 1];
    }
}
matrix multi(matrix x, matrix y)
{
    matrix t;
    int i, j, k;
    memset(t.mat, 0, sizeof(t.mat));
    t.r = x.r;
    t.c = y.c;
    for(i = 1; i <= x.r; i++)
    {
        for(k = 1; k <= x.c; k++)
            if(x.mat[i][k])
            {
                for(j = 1; j <= y.c; j++)
                {
                    t.mat[i][j] += (x.mat[i][k] * y.mat[k][j]) % 10000;
                    t.mat[i][j] %= 10000;
                }
            }
    }
//    for(i=1;i<=2;i++)
//    {
//        for(j=1;j<=2;j++)
//        {
//            for(k=1;j<=2;k++)
//            {
//                t.mat[i][j]+=(x.mat[i][k]*y.mat[k][j])000;
//                t.mat[i][j]%=10000;
//            }
//        }
//    }
    return t;
}
int main()
{
    freopen("in.txt","r",stdin);
    int m;
    while(scanf("%d", &m) != EOF)
    {
        if(m == -1) break;
        init();
        while(m)
        {
            if(m & 1)
            {
                res = multi(origin, res);
            }
            origin = multi(origin, origin);
            m = m >> 1;
        }
        printf("%d\n", res.mat[1][2] % 10000);
    }
    return 0;
}

 

【矩阵加速】 矩阵 快速幂

标签:style   blog   color   io   os   for   sp   div   问题   

原文地址:http://www.cnblogs.com/balfish/p/4014554.html

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