初学矩阵快速幂

洛谷 P3390

#include<bits/stdc++.h>
using namespace std;
#define maxn 300005
#define ll long long
const int mod = 1e9+7;

int n;
ll k;
struct Mat{
	ll m[105][105]; 
	Mat(){
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
				m[j][i]=0;
	}
	inline void build(){
        for(int i=1;i<=n;i++)m[i][i]=1;
    }
}a;

Mat Mul(Mat x,Mat y)
{
    Mat c;
    for(int i=1;i<=n;i++)
    	for(int j=1;j<=n;j++)
    		for(int k=1;k<=n;k++)
				c.m[i][j]=(c.m[i][j]%mod+x.m[i][k]*y.m[k][j]%mod)%mod;
    return c; 
}

Mat poww(Mat x,ll y){
	Mat ans;ans.build();
	while(y){
		if(y&1)ans=Mul(ans,x);
		x=Mul(x,x);
		y>>=1;
	}
	return ans;
}

int main()
{
	scanf("%d %lld",&n,&k);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			scanf("%lld",&a.m[j][i]);
	Mat sum=poww(a,k);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++)
			printf("%lld ",sum.m[j][i]);
		printf("\n");
	}
	return 0;
}

POJ 3070

#include<iostream>
#include<cstdio>
#include<cstring>

using namespace std;
#define maxn 300005
#define ll long long
const int mod = 10000;

ll n;
struct Mat{
	ll m[3][3];
	Mat(){
		memset(m,0,sizeof(m));
	}
	inline void build(){
		for(int i=1;i<=2;i++)m[i][i]=1;
	}
}a;

Mat Mul(Mat x,Mat y){
	Mat c;
	for(int i=1;i<=2;i++)
		for(int j=1;j<=2;j++)
			for(int k=1;k<=2;k++)
				c.m[i][j]=(c.m[i][j]+x.m[i][k]*y.m[k][j]%mod)%mod;
	return c;
}

Mat poww(Mat x,ll y){
	Mat aa;aa.build();
	while(y){
		if(y&1)aa=Mul(aa,x);
		x=Mul(x,x);
		y>>=1;
	}
	return aa;
}

int main()
{
	while(~scanf("%lld",&n)){
		if(n==-1)break;
		if(n==0)printf("0\n");
		else if(n==1||n==2)printf("1\n");
		else{
			a.m[1][1]=1;       //构建矩阵
			a.m[2][1]=1;
			a.m[1][2]=1;
			a.m[2][2]=0;
			Mat ans=poww(a,n-1);
			printf("%lld\n",ans.m[1][1]);
		}
	}
	return 0;
}