矩阵快速幂（构造）

最新推荐文章于 2025-01-25 12:08:59 发布

wwwlps

最新推荐文章于 2025-01-25 12:08:59 发布

阅读量388

点赞数

分类专栏：数论

本文链接：https://blog.youkuaiyun.com/wwwlps/article/details/84196030

版权

数论专栏收录该内容

11 篇文章

订阅专栏

本文深入探讨了矩阵快速幂算法的实现，通过具体代码示例，详细解释了如何使用矩阵快速幂来解决复杂的问题，包括矩阵乘法、矩阵幂运算等关键步骤。文章通过一个竞赛题目，展示了算法的具体应用过程，为读者提供了理解和掌握矩阵快速幂算法的实践指导。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：https://ac.nowcoder.com/acm/contest/221/C

构造矩阵

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <vector>
#include <queue>
using namespace std;
const int INF=0x3f3f3f3f;
const int maxn=10;
const int mod=1e9+7;
typedef long long LL;
LL n;
typedef struct node
{
   LL z[maxn][maxn];
}matrix;
matrix matrix_mul(matrix a,matrix b)
{
   matrix c;
   memset(c.z,0,sizeof(c.z));

   for(int i=1;i<=6;i++)
   {
      for(int j=1;j<=6;j++)
      {
         for(int u=1;u<=6;u++)
         {
            c.z[i][j]+=(a.z[i][u]%mod)*(b.z[u][j]%mod)%mod;
         }
      }
   }
   return c;
}
matrix matrix_pow(matrix a,int n)
{
   matrix ans;
   memset(ans.z,0,sizeof(ans.z));
   for(int i=1;i<=6;i++)
   {
      ans.z[i][i]=1;
   }
   while(n>0)
   {
      if(n&1)
         ans=matrix_mul(ans,a);
      n>>=1;
      a=matrix_mul(a,a);

   }
   return ans;
}
int main()
{
   while(cin>>n)
   {
      if(n==0)
         break;
      matrix t;
      memset(t.z,0,sizeof(t.z));
      t.z[1][1]=1;t.z[1][2]=1;t.z[1][3]=0;t.z[1][4]=1;t.z[1][5]=0;t.z[1][6]=0;
      t.z[2][1]=0;t.z[2][2]=2;t.z[2][3]=3;t.z[2][4]=0;t.z[2][5]=1;t.z[2][6]=1;
      t.z[3][1]=0;t.z[3][2]=1;t.z[3][3]=0;t.z[3][4]=0;t.z[3][5]=0;t.z[3][6]=0;

      t.z[4][1]=0;t.z[4][2]=0;t.z[4][3]=0;t.z[4][4]=1;t.z[4][5]=2;t.z[4][6]=1;
      t.z[5][1]=0;t.z[5][2]=0;t.z[5][3]=0;t.z[5][4]=0;t.z[5][5]=1;t.z[5][6]=1;
      t.z[6][1]=0;t.z[6][2]=0;t.z[6][3]=0;t.z[6][4]=0;t.z[6][5]=0;t.z[6][6]=1;

      matrix y;
      y.z[1][1]=2;
      y.z[2][1]=2;
      y.z[3][1]=1;
      y.z[4][1]=4;
      y.z[5][1]=2;
      y.z[6][1]=1;
      matrix zz=matrix_pow(t,n-1);
      LL res=0;
     // for(int i=1;i<=6;i++)
       //  cout << ans.z[1][i] << ' ' ;
      //cout << endl;
      for(int i=1;i<=6;i++)
      {
         res+=zz.z[1][i]*y.z[i][1];
         res%=mod;
      }
      cout  << res << endl;

   }
   return 0;
}