题目
问题描述
给定一个矩阵A,一个非负整数b和一个正整数m,求A的b次方除m的余数。
其中一个nxn的矩阵除m的余数得到的仍是一个nxn的矩阵,这个矩阵的每一个元素是原矩阵对应位置上的数除m的余数。
要计算这个问题,可以将A连乘b次,每次都对m求余,但这种方法特别慢,当b较大时无法使用。下面给出一种较快的算法(用A^b表示A的b次方):
若b=0,则A^b%m=I%m。其中I表示单位矩阵。
若b为偶数,则A^b%m=(A^(b/2)%m)^2%m,即先把A乘b/2次方对m求余,然后再平方后对m求余。
若b为奇数,则A^b%m=(A^(b-1)%m)*a%m,即先求A乘b-1次方对m求余,然后再乘A后对m求余。
这种方法速度较快,请使用这种方法计算A^b%m,其中A是一个2x2的矩阵,m不大于10000。
输入格式
输入第一行包含两个整数b, m,第二行和第三行每行两个整数,为矩阵A。
输出格式
输出两行,每行两个整数,表示A^b%m的值。
样例输入
2 2
1 1
0 1
样例输出
1 0
0 1
自己先开始看到魔方那题,准备用套路求,但套路不出来
我先开始准备用这个方法直接求得,但是发现有问题
import java.util.Scanner;
public class 矩阵的乘方 {
private static int n;
private static int[][] a;
public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner sc=new Scanner(System.in);
n = sc.nextInt();
a = new int[n][n];
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
a[i][j]=sc.nextInt();
}
}
int[][] ass=new int[n][n];
ass=pow(a,2);
for(int i=0;i<n;i++){
for(int j=0;j<n;j++)
System.out.print(ass[i][j]+" ");
System.out.println();
}
}
public static int[][] multy(int[][] a,int[][] b){
int[][] ass=new int[n][n];
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
ass[i][j]=0;
for(int d=0;d<n;d++){
a[i][j]+=a[i][d]*a[d][j];
}
}
}
return ass;
}
public static int[][] pow(int[][] a,int d){
int[][] ass=new int[n][n];
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
if(i==j)
ass[i][j]=1;
else
ass[i][j]=0;
}
}
while(d!=0){
if((d&1)==1){
ass=multy(ass,a);
}
a=multy(a,a);
d>>=1;
}
return ass;
}
}
import java.util.Scanner;
public class 矩阵乘方1 {
private static int m;
public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner sc=new Scanner(System.in);
int b=sc.nextInt();
m = sc.nextInt();
int[][] a=new int[2][2];
for(int i=0;i<2;i++){
for(int j=0;j<2;j++){
a[i][j]=sc.nextInt();
}
}
int[][] ass=new int[2][2];
if(b==0){
ass=pow(a,0);
}
if((b&1)!=1){
ass=f(pow(a,b-1));
ass=f(pow(ass,2));
}
if((b&1)==1){
ass=f(pow(a,b/2));
ass=f(multy(ass,a));
}
for(int i=0;i<2;i++){
for(int j=0;j<2;j++){
System.out.println(ass[i][j]+" ");
}
System.out.println();
}
}
public static int[][] multy(int[][] a,int[][] b){
int[][] ass=new int[2][2];
for(int i=0;i<2;i++){
for(int j=0;j<2;j++){
ass[i][j]=0;
for(int d=0;d<2;d++){
a[i][j]+=a[i][d]*a[d][j];
}
}
}
return ass;
}
public static int[][] pow(int[][] a,int d){
int[][] ass=new int[2][2];
for(int i=0;i<2;i++){
for(int j=0;j<2;j++){
if(i==j)
ass[i][j]=1;
else
ass[i][j]=0;
}
}
if(d==0)
return ass;
while(d!=0){
if((d&1)==1){
ass=multy(ass,a);
}
a=multy(a,a);
d>>=1;
}
return ass;
}
public static int[][] f(int[][] a){
for(int i=0;i<2;i++){
for(int j=0;j<2;j++){
a[i][j]%=m;
}
}
return a;
}
}