51nod 1113 矩阵快速幂

矩阵快速幂板子

#include <bits/stdc++.h>

typedef long long ll;
typedef long long ld;

using namespace std;

const ll maxn = 1e5 + 7;

ll n,m;
struct node
{
	//矩阵 
	ll matrix[105][105];
};

//矩阵乘法 
node cal(node a,node b)
{
	ll k = 1;
	node c = {0};
	for(int i = 1; i <= n; i ++)
	{
		for(int j = 1; j <= n; j ++)
		{
			while(k <= n)
			{
				c.matrix[i][j]  += a.matrix[i][k] * b.matrix[k][j];
				k++;
			}
			k = 1;
		}
	}
	return c;
}

//矩阵快速幂
//a代表一个矩阵，m代表几次幂 
node matqp(node a,ll m)
{
	//c是单位矩阵 
	node c = {0};
	for(int i = 1; i <= n; i ++)
	{
		c.matrix[i][i] = 1;
	}
	
	while(m)
	{
		if(m & 1)c = cal(c,a);
		m >>= 1;
		a = cal(a,a);
	}
	
	return c;
} 
 
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0),cout.tie(0);
	
	node c;
	cin >> n >> m;
	for(int i = 1; i <= n; i ++)
	{
		for(int j = 1; j <= n; j ++)
		{
			cin >> c.matrix[i][j];
		}
	}
	
	c = matqp(c,m);
	
	for(int i = 1; i <= n; i ++)
	{
		for(int j = 1; j <= n; j ++)
		{
			cout << c.matrix[i][j] << " ";
		}
		cout << endl;
	}
	
	return 0;
}

 

1113 矩阵快速幂

基准时间限制：3 秒 空间限制：131072 KB 分值: 40 难度：4级算法题

收藏

 

关注

给出一个N * N的矩阵，其中的元素均为正整数。求这个矩阵的M次方。由于M次方的计算结果太大，只需要输出每个元素Mod (10^9 + 7）的结果。

Input

第1行：2个数N和M，中间用空格分隔。N为矩阵的大小，M为M次方。(2 <= N <= 100, 1 <= M <= 10^9)
第2 - N + 1行：每行N个数，对应N * N矩阵中的1行。(0 <= N[i] <= 10^9)

Output

共N行，每行N个数，对应M次方Mod (10^9 + 7)的结果。

Input示例

2 3
1 1
1 1

Output示例

4 4
4 4

 

#include <bits/stdc++.h>
typedef long long ll;
const ll mod=1e9+7;
using namespace std;
ll n,m;
struct kk
{
    ll a[105][105];
}
t;
kk juzhen(kk a,kk b)
{
    kk c={0};
    for(ll i=0;i<n;i++)
    {
        for(ll j=0;j<n;j++)
        {
            for(int k = 0 ; k < n ; k ++)
                 c.a[i][j]=(c.a[i][j]+a.a[i][k]*b.a[k][j])%mod;
        }
    }
    return c;
}
kk kuaisu(ll m)
{
    kk b;
    memset(b.a,0,sizeof(b.a));
    for(ll i=0;i<n;i++)
    {
        b.a[i][i]=1;
    }
    if(n<0)return b;
    while(m)
    {
        if(m&1)b=juzhen(b,t);
        m>>=1;
        t=juzhen(t,t);
    }
    return b;
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>n>>m;
    for(ll i=0;i<n;i++)
    {
        for(ll j=0;j<n;j++)
        {
            cin>>t.a[i][j];
        }
    }
    kk dd=kuaisu(m);
    for(ll i=0;i<n;i++)
    {
        for(ll j=0;j<n;j++)
        {
            cout<<dd.a[i][j]<<' ';
        }
        cout<<endl;
    }
    return 0;
}