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

矩阵快速幂

时间:2018-09-18 14:23:46      阅读:150      评论:0      收藏:0      [点我收藏+]

标签:img   memcpy   src   splay   inline   isp   快速   递推   const   

线性递推可用矩阵快速幂\(O(\log nk^3)\)解。

构造系数矩阵的方法是先列出转移式,然后看哪些项是要的填为1,否则填0。如对\[f_{i,0}=f_{i-1,1}\] \[f_{i,1}=f_{i-1,0}\]

则构造矩阵技术分享图片

矩阵快速幂代码如下:

const int D[6][6] = {{1, 1, 1, 1, 1, 0},
                     {1, 0, 0, 1, 0, 0},
                     {1, 0, 0, 0, 0, 1},
                     {1, 1, 0, 0, 0, 0},
                     {1, 0, 0, 0, 0, 0},
                     {0, 0, 1, 0, 0, 0}};
const int F[6] = {1, 0, 0, 0, 0, 0};
int n;

int d[6][6], f[6], g[6], gd[6][6];
inline void calcf() {
    memset(g, 0, sizeof(g));
    for(int i = 0; i < 6; i++) for(int j = 0; j < 6; j++) Sum(g[i], prod(f[j], d[i][j]));
    memcpy(f, g, sizeof(g));
}
inline void calcd() {
    memset(gd, 0, sizeof(gd));
    for(int k = 0; k < 6; k++) for(int i = 0; i < 6; i++) for(int j = 0; j < 6; j++) Sum(gd[i][j], prod(d[i][k], d[k][j]));
    memcpy(d, gd, sizeof(gd));
}

int main() {
    while(n) {
        if(n&1) calcf();
        calcd();
        n >>= 1;
    }
    // ...
}

矩阵快速幂

标签:img   memcpy   src   splay   inline   isp   快速   递推   const   

原文地址:https://www.cnblogs.com/utopia999/p/9668491.html

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