Matrix Power Series【矩阵乘法】

最新推荐文章于 2024-04-29 09:13:48 发布

原创最新推荐文章于 2024-04-29 09:13:48 发布 · 238 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

矩阵乘法同时被 2 个专栏收录

17 篇文章

订阅专栏

快速幂

16 篇文章

订阅专栏

Time Limit: 3000MS Memory Limit: 131072K
Total Submissions: 35006 Accepted: 14029

Description

Given a $n \times n$ matrix A and a positive integer k, find the sum $S = A + A 2 + A 3 + \dots + A k$ .

Input

The input contains exactly one test case. The first line of input contains three positive integers n (n ≤ 30), k (k ≤ 109) and m (m < 104). Then follow n lines each containing n nonnegative integers below 32,768, giving A’s elements in row-major order.

Output

Output the elements of S modulo m in the same way as A is given.

Sample Input

2 2 4
0 1
1 1

Sample Output

1 2
2 3

题目翻译

给定一个 $n \times n$ 矩阵a和一个正整数k，求和 $S = a + A 2 + A 3 + \dots + A k$ 。

输入
只包含一个测试用例。输入的第一行包含三个正整数 $n（n≤30）、k（k≤10^9）和m（m<10^4）$ 。然后跟随n行，每行包含 $32768$ 以下的n个非负整数，按照行的主要顺序给出A的元素。

输出
以与A相同的方式输出S模m的元素。

解题思路

矩阵乘法

考虑 $1 \times 2$ 的矩阵 $A^n-1,S[n-2]】$ ，注意此1×2矩阵的2个元素都是r阶方阵！我们希望通过乘以某2×2的矩阵M，得到1×2的矩阵 $A^n,S[n-1]】＝【A^n-1*A, An-1+S[n-2]】$
容易构造出此矩阵M为：
在这里插入图片描述
其中4个元素均为r阶方阵，O表示r阶全0矩阵，E表示单位矩阵（主对角线上为1，其它全为0）。
问题解决。我们的复杂度是: $(2 r) 3 * l o g n$

其实，这个矩阵就是那个流行方法中用到的，这里我们用前面几个问题的思想很容易构造出了这个 $2 r * 2 r$ 的矩阵。因此：此思想高效、一般、统一、和谐！

代码

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;
long long nn,k,mm;
struct c{
	long long n,m;
	long long a[100][100];
}A,B,CC;
c operator *(c A,c B){
	c C;
	C.n=A.n,C.m=B.m;
	for(int i=1;i<=C.n;i++)
		for(int j=1;j<=C.m;j++)
			C.a[i][j]=0;
	for(int k=1;k<=A.m;k++)
	{
		for(int i=1;i<=C.n;i++)
			for(int j=1;j<=C.m;j++)
				C.a[i][j]=(C.a[i][j]+(A.a[i][k]*B.a[k][j])%mm)%mm;
	}	
	return C;
}
void poww(long long x){
	if(x==1)
	{
		B=A;
		return; 
	}
	poww(x>>1);
	B=B*B;
	if(x&1)
		B=B*A;
	
}
int main(){
	scanf("%lld%lld%lld",&nn,&k,&mm);
	CC.n=nn,CC.m=2*nn;
	A.n=2*nn,A.m=2*nn;
	for(int i=1;i<=nn;i++)
		for(int j=1;j<=nn;j++)
		{
			scanf("%lld",&CC.a[i][j]);
			CC.a[i][j]=CC.a[i][j]%mm;
			A.a[i][j]=CC.a[i][j];
		}	
	for(int i=1;i<=nn;i++)
		for(int j=nn+1;j<=2*nn;j++)
			if(j-nn==i)
				A.a[i][j]=1;	
	for(int i=nn+1;i<=2*nn;i++)
		for(int j=nn+1;j<=2*nn;j++)
			if(j==i)
				A.a[i][j]=1;
	poww(k);
	CC=CC*B;
	
	for(int i=1;i<=nn;i++)
	{
		for(int j=nn+1;j<=2*nn;j++)
			printf("%lld ",CC.a[i][j]);
		printf("\n");
	}		
}