HDU1575(矩阵快速幂模板题)

37 篇文章 0 订阅
8 篇文章 0 订阅

简单的矩阵快速幂,输入矩阵直接套模板做就行了。
code

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<iostream>
#include<string>
#include <set>
#include<time.h>
//a&3==a%4
using namespace std ;
#define ll long long
#define mem(a) memset(a,0,sizeof(a))
const double eps = 1e-8;
const int maxn = 110010;
const int inf = 0x3f3f3f3f;
const ll mod=9973ll;

struct matrix
{
    ll mat[10][10];//mat[i][j]=µÚiÐеÚjÁÐ
    matrix()
    {
        for(int i=0;i<10;i++)
        {
            for(int j=0;j<10;j++)
            {
                mat[i][j]=0ll;
            }
        }
    }
};

void out(matrix a,int n)
{
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            cout<<a.mat[i][j]<<"   ";
        }
        cout<<endl;
    }
}

matrix multiply(matrix a,matrix b,int n)
{
    matrix res;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            for(int k=0;k<n;k++)
            {
                res.mat[i][j]+=a.mat[i][k]*b.mat[k][j]%mod;
                res.mat[i][j]%=mod;
            }
        }
    }
    return res;
}

matrix quickmi(matrix a,ll b,int n)
{
    matrix E;
    matrix res;
    for(int i=0;i<n;i++)
    {
        E.mat[i][i]=1;
        res.mat[i][i]=1;
    }
    E=multiply(a,E,n);
    while(b>0)
    {
        if(b%2==1)
        {
            res=multiply(res,E,n);
        }
        b=b/2;
        E=multiply(E,E,n);
    }
    return res;
}

int main()
{;
    int _case;
    scanf("%d",&_case);
    while(_case--)
    {
        int n;
        ll k;
        matrix sum;
        scanf("%d",&n);
        cin>>k;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                ll a;
                cin>>a;
                sum.mat[i][j]=a;
            }
        }
        matrix res=quickmi(sum,k,n);
        ll sumres=0;
        for(int i=0;i<n;i++)
        {
            sumres=(sumres+res.mat[i][i])%mod;
        }
        cout<<sumres%mod<<endl;
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值