hdu1575 二分+矩阵求幂

矩阵M^k,求幂,如果k是偶数,M^k=M^(k/2)*M*(k/2)%mod; 如果k是奇数,M^k=M^(k/2)*M*(k/2)*M;采用二进制扫描
#include<iostream>
#include<stdio.h>
#include<string>
#include<string.h>
#define N 11
#define mod 9973
using namespace std;
struct Matrix
{
    int mat[N][N];

};
int n;
Matrix Mat_Mul(Matrix M1,Matrix M2) { Matrix c; for(int i=0;i<n;i++) { for(int j=0;j<n;j++) { c.mat[i][j]=0; for(int k=0;k<n;k++) { c.mat[i][j]+=(M1.mat[i][k]*M2.mat[k][j])%mod; } c.mat[i][j]=c.mat[i][j]%mod; } } return c; } Matrix Bi_Search(Matrix M,int k) { Matrix E; memset(E.mat,0,sizeof(E.mat)); for(int i=0;i<n;i++) { E.mat[i][i]=1; } while(k) { if(k&1) E=Mat_Mul(E,M); M=Mat_Mul(M,M); k>>=1; } return E; } int main() { int t; cin>>t; while(t--) { int k; Matrix M; int Tr=0; cin>>n>>k; for(int i=0;i<n;i++) { for(int j=0;j<n;j++) { cin>>M.mat[i][j]; } } M=Bi_Search(M,k); for(int i=0;i<n;i++) { Tr+= M.mat[i][i]; } cout<<Tr%mod<<endl; } return 0 ; }
原文地址:https://www.cnblogs.com/zn505119020/p/3552998.html