矩阵快速幂模板（pascal）

洛谷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

算法：矩阵快速幂

 

矩阵快速幂模板：

program rrr(input,output);
const
  cs=1000000007;
var
  a,c,ans:array[0..110,0..110]of int64;
  n,i,j,k:longint;
  m:int64;
begin
   assign(input,'r.in');assign(output,'r.out');reset(input);rewrite(output);
   readln(n,m);
   for i:=1 to n do for j:=1 to n do read(a[i,j]);
   fillchar(ans,sizeof(ans),0);
   for i:=1 to n do ans[i,i]:=1;
   while m>0 do
      begin
         if m mod 2=1 then
            begin
               //c:=ans*a;
               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;
               //ans:=c;
               for i:=1 to n do for j:=1 to n do ans[i,j]:=c[i,j];
            end;
         //c:=a*a;
         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;
         //a:=c;
         for i:=1 to n do for j:=1 to n do a[i,j]:=c[i,j];
         m:=m>>1;
      end;
   for i:=1 to n do begin for j:=1 to n do write(ans[i,j],' ');writeln; end;
   close(input);close(output);
end.