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

矩阵快速幂 模板

时间:2018-04-15 17:15:00      阅读:147      评论:0      收藏:0      [点我收藏+]

标签:namespace   运算   color   pre   const   ret   cst   turn   names   

在矩阵快速幂中要注意可以把两个矩阵化为同大小的时候运算

 

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;

const int MAX_N=62;
int n;

struct JuZhen
{
    int m[MAX_N][MAX_N];
}a,b;

JuZhen Mul(JuZhen x,JuZhen y)
{
      JuZhen sum;
      memset(sum.m,0,sizeof(sum.m));
      for(int k=0;k<n;k++)
      {
          for(int i=0;i<n;i++)
          {
              if(!x.m[i][k])
                continue;
              for(int j=0;j<n;j++)
              {
                  sum.m[i][j]=(sum.m[i][j]+x.m[i][k]*y.m[k][j]);
              }
          }
      }
      return sum;
}

JuZhen fastm(JuZhen a,int b)
{
    JuZhen s;
    memset(s.m,0,sizeof(s.m));
    for(int i=0;i<n;i++)
    {
        s.m[i][i]=1;
    }
    while(b)
    {
        if(b&1)
          s=Mul(s,a);    
        a=Mul(a,a);
        b=b>>1;

}
    return s;
}

int main()
{
    memset(a.m,0,sizeof(b.m));
    memset(b.m,0,sizeof(b.m));
    cin>>n;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            cin>>a.m[i][j];
        }
    }
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            cin>>b.m[i][j];
        }
    }
}

 

矩阵快速幂 模板

标签:namespace   运算   color   pre   const   ret   cst   turn   names   

原文地址:https://www.cnblogs.com/Yinchen-One/p/8848225.html

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