首页 > 代码库 > 矩阵快速幂模板

矩阵快速幂模板

2024-09-05 00:50:53   219人阅读

洛谷P3390

题目背景

矩阵快速幂

题目描述

给定n*n的矩阵A，求A^k

输入输出格式

输入格式：

 

第一行，n,k

第2至n+1行，每行n个数，第i+1行第j个数表示矩阵第i行第j列的元素

 

输出格式：

 

输出A^k

共n行，每行n个数，第i行第j个数表示矩阵第i行第j列的元素，每个元素模10^9+7

 

输入输出样例

输入样例#1：
2 1
1 1
1 1
输出样例#1：
1 1
1 1

说明

n<=100, k<=10^12, |矩阵元素|<=1000

算法：矩阵快速幂

 

矩阵快速幂模板：

 1 program rrr(input,output);
 2 const
 3   cs=1000000007;
 4 var
 5   a,c,ans:array[0..110,0..110]of int64;
 6   n,i,j,k:longint;
 7   m:int64;
 8 begin
 9    assign(input,‘r.in‘);assign(output,‘r.out‘);reset(input);rewrite(output);
10    readln(n,m);
11    for i:=1 to n do for j:=1 to n do read(a[i,j]);
12    fillchar(ans,sizeof(ans),0);
13    for i:=1 to n do ans[i,i]:=1;
14    while m>0 do
15       begin
16          if m mod 2=1 then
17             begin
18                //c:=ans*a;
19                for i:=1 to n do for j:=1 to n do begin c[i,j]:=0;for k:=1 to n do c[i,j]:=(c[i,j]+ans[i,k]*a[k,j]) mod cs; end;
20                //ans:=c;
21                for i:=1 to n do for j:=1 to n do ans[i,j]:=c[i,j];
22             end;
23          //c:=a*a;
24          for i:=1 to n do for j:=1 to n do begin c[i,j]:=0;for k:=1 to n do c[i,j]:=(c[i,j]+a[i,k]*a[k,j]) mod cs; end;
25          //a:=c;
26          for i:=1 to n do for j:=1 to n do a[i,j]:=c[i,j];
27          m:=m>>1;
28       end;
29    for i:=1 to n do begin for j:=1 to n do write(ans[i,j],‘ ‘);writeln; end;
30    close(input);close(output);
31 end.

 

矩阵快速幂模板


联系
我们