【矩阵快速幂模板】

来源:互联网 发布:淘宝好看的小白鞋 编辑:程序博客网 时间:2024/06/18 05:21


原文来自:点击打开链接


下面是 m^n  % k 的快速幂:

// m^n k
int quickpow(int m,int n,int k)
{
    int 1;
    while (n 0)
    {
          if (n 1)
             (b*m)%k;
          >> ;
          (m*m)%k;
    }
    return b;


下面是矩阵快速幂:
//HOJ 3493

|| 快速幂(quickpow)模板 
|| 为等比,I 
为单位矩阵
|| MAX 初始化!!!!
||


#include 
const int MAX 3;

typedef  struct{
        int  m[MAX][MAX];
 Matrix;

Matrix {5,-7,4,
            1,0,0,
            0,1,0,
           };

Matrix {1,0,0,
            0,1,0,
            0,0,1,
           };
           
Matrix matrixmul(Matrix a,Matrix b) //矩阵乘法
{
       int i,j,k;
       Matrix c;
       for (i MAX; i++)
           for (j 0; MAX;j++)
             {
                 c.m[i][j] 0;
                 for (k 0; MAX; k++)
                     c.m[i][j] += (a.m[i][k] b.m[k][j])�97;
                 c.m[i][j] %= 9997;
             }
       return c;
}
          
Matrix quickpow(long long n)
{
       Matrix P, I;
       while (n >= 1)
       {
             if (n 1)
                matrixmul(b,m);
             >> 1;
             matrixmul(m,m);
       }
       return b;
}
               


int main()
{
    Matrix re;
    int f[3] {2,6,19};
    long long n;
    while (scanf("%I64d",&n) && != 0)
    {
          if (n == 1)
             printf("1\n");
          else if (n <= 4)
                  printf("%d\n",f[n-2]);
               else {
                      re quickpow(n 4);
                      printf("%d\n",(((re.m[0][0]*f[2]) 
                             (re.m[0][1]*f[1]) (re.m[0][2]*f[0])) �97 9997) 9997);
                      }
    }
    return 0;
}
原创粉丝点击