蓝桥杯 基础练习(17) 矩阵乘法(C语言)

问题描述
　　给定一个N阶矩阵A，输出A的M次幂（M是非负整数）
　　例如：
　　A =
　　1 2
　　3 4
　　A的2次幂
　　7 10
　　15 22
输入格式
　　第一行是一个正整数N、M（1<=N<=30, 0<=M<=5），表示矩阵A的阶数和要求的幂数
　　接下来N行，每行N个绝对值不超过10的非负整数，描述矩阵A的值
输出格式
　　输出共N行，每行N个整数，表示A的M次幂所对应的矩阵。相邻的数之间用一个空格隔开

#include<stdio.h>
#include<string.h>
int main()
{
    int N, M ;
    scanf("%d%d",&N,&M) ;
    int a[30][30] ;
    int b[30][30] ;
    int c[30][30] ;
    for(int i=0; i<N; i++){
        for(int j=0; j<N ;j++){
            scanf("%d",&a[i][j]) ;
            b[i][j] = a[i][j];
        }
    }
    if(M == 0){
        for(int i=0; i<N; i++){
            for(int j=0; j<N; j++){
                if(i == j)
                    b[i][j] = 1;
                else
                    b[i][j] = 0;
            }
        }
    }else{
        int m = M-1;
        while(m--){
            memset(c, 0, sizeof(c));
            for(int i=0; i<N; i++){
                for(int j=0; j<N; j++){
                    for(int k=0; k<N; k++){
                        c[i][j] += b[i][k]*a[k][j];
                    }
                }
            }
            for(int i=0; i<N; i++)
                for(int j=0; j<N; j++)
                    b[i][j] = c[i][j];
       } 
    }
    for(int i=0; i<N; i++){
    	for(int j=0; j<N; j++){
        	printf("%d",b[i][j]) ;
        	if(j!=N-1)
        	  printf(" ") ;
		}
		printf("\n") ;
	}       
    return 0;
}