矩阵乘法

Description

给定一个N阶矩阵A，输出A的M次幂（M是非负整数）
　　例如：
　　A =
　　1 2
　　3 4
　　A的2次幂
　　7 10
　　15 22

Input

第一行是一个正整数N、M（1<=N<=30, 0<=M<=5），表示矩阵A的阶数和要求的幂数。

接下来N行，每行N个绝对值不超过10的非负整数，描述矩阵A的值。

Output

输出共N行，每行N个整数，表示A的M次幂所对应的矩阵。相邻的数之间用一个空格隔开。

 

Sample Input

2 2
1 2
3 4

Sample Output

7 10
15 22

#include<iostream>
using namespace std;
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    int a[100][100],b[100][100],c[100][100];
    for(int i=1;i<=n;i++){//输入a矩阵时，b矩阵也赋一样的值 
        for(int j=1;j<=n;j++){
            scanf("%d",&a[i][j]);
            b[i][j]=a[i][j];
        }
    }
    if(m==0){//如果是零次幂，就输出单位矩阵 
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                if(i==j)
                    c[i][j]=1;
                else
                    c[i][j]=0;    
                printf("%d ",c[i][j]);
            }
        }
        printf("
");
        return 0;
    }
    if(m==1){//如果是1次幂，就输出自身 
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                printf("%d ",a[i][j]);
            }
            printf("
");
        }
        return 0;
    }
    //如果是多次幂 
    for(int time=1;time<m;time++){//如果是m次幂，就进行m-1次乘法 
        for(int i=1;i<=n;i++){//矩阵的行 
            for(int j=1;j<=n;j++){//矩阵的列 
                int sum=0;//用来盛放一行乘一列的总和 
                for(int k=1;k<=n;k++){//一行乘一列 
                    sum+=a[i][k]*b[k][j];
                }
                c[i][j]=sum;//一行乘一列完后，赋值给数组c相应位置 
            }
        }
        for(int i=1;i<=n;i++){//完成一次乘法后 
            for(int j=1;j<=n;j++){//将c数组复制到b数组 
                b[i][j]=c[i][j];//b数组再继续和a数组相乘 
            }
        }
    }
    for(int i=1;i<=n;i++){//最后输出c数组 
        for(int j=1;j<=n;j++){
            printf("%d ",c[i][j]);
        }
        printf("
");
    }
    return 0;
}

首先得知道矩阵怎样作乘法，核心代码是这个

    for(int time=1;time<m;time++){//如果是m次幂，就进行m-1次乘法 
        for(int i=1;i<=n;i++){//矩阵的行 
            for(int j=1;j<=n;j++){//矩阵的列 
                int sum=0;//用来盛放一行乘一列的总和 
                for(int k=1;k<=n;k++){//一行乘一列 
                    sum+=a[i][k]*b[k][j];
                }
                c[i][j]=sum;//一行乘一列完后，赋值给数组c相应位置 
            }
        }
        for(int i=1;i<=n;i++){//完成一次乘法后 
            for(int j=1;j<=n;j++){//将c数组复制到b数组 
                b[i][j]=c[i][j];//b数组再继续和a数组相乘 
            }
        }
    }

理解好这个！

O(∩_∩)O