码迷,mamicode.com
首页 > 其他好文 > 详细

luogu3390 【模板】矩阵快速幂

时间:2017-12-11 14:27:10      阅读:119      评论:0      收藏:0      [点我收藏+]

标签:一个   gpo   code   down   source   while   struct   scanf   cin   

#include <iostream>
#include <cstdio>
using namespace std;
typedef long long ll;
ll k;
const ll mod=1e9+7;
struct Matrix{
    int n;
    ll num[105][105];
    Matrix operator *(const Matrix &b)const{
        Matrix res=b;
        for(int i=1; i<=n; i++)
            for(int j=1; j<=n; j++){
                res.num[i][j] = 0;
                for(int k=1; k<=n; k++)
                    res.num[i][j] = (res.num[i][j]+num[i][k]*b.num[k][j]%mod)%mod;
            }
        return res;
    }
    Matrix operator ^(ll k)const{
        Matrix res;
        Matrix x = *this;
        res.n = n;
        for(int i=1; i<=n; i++)
            for(int j=1; j<=n; j++)
                if(i==j)    res.num[i][j] = 1;
                else        res.num[i][j] = 0;//搞成一个单位矩阵,单位矩阵乘矩阵A等于矩阵A本身。
        while(k){
            if(k&1) res = res * x;
            x = x * x;
            k >>= 1;
        }
        return res;
    }
}a;
int main(){
    cin>>a.n>>k;
    for(int i=1; i<=a.n; i++)
        for(int j=1; j<=a.n; j++)
            scanf("%lld", &a.num[i][j]);
    a = a ^ k;
    for(int i=1; i<=a.n; i++){
        for(int j=1; j<=a.n; j++)
            printf("%lld ", a.num[i][j]);
        printf("\n");
    }
    return 0;
}

luogu3390 【模板】矩阵快速幂

标签:一个   gpo   code   down   source   while   struct   scanf   cin   

原文地址:http://www.cnblogs.com/poorpool/p/8022160.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!