矩陣快速冪

http://acm.hdu.edu.cn/showproblem.php?pid=1575

#include <iostream>
#include <string.h>
#include <stdio.h>
using namespace std;
const int N=11,mod=9973;
int n,b;
struct Matrix
{
    int m[N][N];
}curr;
Matrix multi(Matrix a,Matrix b)
{
    Matrix ans;
    memset(ans.m,0,sizeof(ans.m));
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            for(int k=0;k<n;k++)
            {
                ans.m[i][j]=(ans.m[i][j]+a.m[i][k]*b.m[k][j])%mod;
            }
        }
    }
    return ans;
}
Matrix fast_pow(Matrix a,int b)
{
    Matrix ans,base=a;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            ans.m[i][j]=(i==j);
        }
    }
    while(b)
    {
        if(b&1)
        {
            ans=multi(ans,base);
        }
        base=multi(base,base);
        b>>=1;
    }
    return ans;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&b);
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                scanf("%d",&curr.m[i][j]);
            }
        }
        curr=fast_pow(curr,b);
        int ans=0;
        for(int i=0;i<n;i++)
        {
            ans=(ans+curr.m[i][i])%mod;
        }
        printf("%d\n",ans%mod);
    }

}
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容