[ロゴP3390] [クイックマトリックスパワー] [テンプレート]

タイトル説明

N×N行列Aが与えられると、検索^Kを

入力フォーマット

最初の行に2つの整数n、k、次の行にn行、各行にn個の整数、i番目の行のj番目の数値はAi、jを表します

出力フォーマット

出力^Ak

合計n行あり、各行にn個の数値があります。i番目の行のj番目の数値はA ^k i、jを表し、各要素は10 ⁹ + 7を法とします。

サンプルの入力と出力

＃1を入力してください

2 1
1 1
1 1
1
2
3

出力＃1

1 1
1 1
1
2

分析：

行列の乗算と行列の高速パワーのテンプレート問題の
鍵は、" ∗*を再定義することです。$\ast$ "。
乗算記号を行列乗算として定義します。
次に、直接呼び出します。

コードをアップロード

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;

ll n,k;
const int mod=1000000007;

struct matrix
{
    
    
	ll f[101][101];
}A,B;

matrix operator *(matrix a,matrix b)
{
    
    
	matrix C;
	for(int i=1;i<=n;i++)
	{
    
    
		for(int j=1;j<=n;j++)
		{
    
    
			C.f[i][j]=0;
		}
	}
	for(int k=1;k<=n;k++)
	{
    
    
		for(int i=1;i<=n;i++)
		{
    
    
			for(int j=1;j<=n;j++)
			{
    
    
				C.f[i][j]=(C.f[i][j]+a.f[i][k]*b.f[k][j]%mod)%mod;
			}
		}
	}
	return C;
}

void ksm(ll x)
{
    
    
	if(x==1)
	{
    
    
		B=A;
		return;
	}
	ksm(x/2);
	B=B*B;
	if(x&1) B=B*A;
}

int main()
{
    
    
    cin>>n>>k; 
    for(int i=1;i<=n;i++)
	{
    
    
		for(int j=1;j<=n;j++)
		{
    
    
			scanf("%lld",&A.f[i][j]);
		}
	}
	ksm(k);
	for(int i=1;i<=n;i++)
	{
    
    
		for(int j=1;j<=n;j++)
		{
    
    
			cout<<B.f[i][j]<<' ';
		}
		cout<<endl;
	 } 
	return 0;
}