java蓝桥杯练习 矩阵乘方

java蓝桥杯练习 矩阵乘方

资源限制
时间限制：1.0s 内存限制：512.0MB
问题描述
　　给定一个矩阵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

//java code
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
    static int m=0;
    public static void main(String[] args) throws IOException {
        BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
        String s[]=br.readLine().split(" ");
        int b=Integer.parseInt(s[0]);
        m=Integer.parseInt(s[1]);
        int re[][]=new int[2][2];
        int ma[][]=new int[2][2];
        for (int i = 0; i < 2; i++) {
            String tag[]=br.readLine().split(" ");
            for (int j = 0; j < 2; j++) {
                ma[i][j]=Integer.parseInt(tag[j]);
            }
        }
        re=powmul(ma,b);
        for (int i = 0; i < 2; i++) {
            for (int j = 0; j < 2; j++) {
                System.out.print(re[i][j]%m+" ");
            }
            System.out.println();
        }
    }
    public static int[][] mul(int x[][],int y[][]){
        int temp[][]=new int[2][2];
        for (int i = 0; i < 2; i++) {
            for (int j = 0; j < 2; j++) {
                for (int k = 0; k < 2; k++) {
                    temp[i][j]+=x[i][k]*y[k][j];
                    temp[i][j]%=m;
                }
            }
        }
        return temp;
    }
    public static int[][] powmul(int a[][],int b){
        if(b==0){
            int temp[][]=new int[2][2];
            for (int i = 0; i < 2; i++) {
                temp[i][i]=1;
            }
            return temp;
        }if(b==1){
            return a;
        }
        int rep[][]=powmul(a,b/2);
        if(b%2==0)
            return mul(rep,rep);
        else
            return mul(mul(rep,rep),a);
    }
}