基础练习 矩阵乘法

  基础练习 矩阵乘法  

时间限制：1.0s   内存限制：512.0MB

      

锦囊1

二维数组，循环。

锦囊2

使用二维数组保存矩阵，乘法使用新的数组保存中间结果。

问题描述

　　给定一个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次幂所对应的矩阵。相邻的数之间用一个空格隔开

样例输入

2 2
1 2
3 4

样例输出

7 10
15 22

思路：

a数组用来存放最开始的矩阵，一直不变

b数组用来存放运算过程的矩阵，注意每一次乘完以后都要memset，为0

c数组用来存放这一次运算以后得到的结果

按矩阵运算用循环模拟即可，

这道题要注意m=0的情况，例如下边这个例子

input

10 0
10 10 10 10 10 10 10 10 10 10 
10 10 10 10 10 10 10 10 10 10 
10 10 10 10 10 10 10 10 10 10 
10 10 10 10 10 10 10 10 10 10 
10 10 10 10 10 10 10 10 10 10 
10 10 10 10 10 10 10 10 10 10 
10 10 10 10 10 10 10 10 10 10 
10 10 10 10 10 10 10 10 10 10 
10 10 10 10 10 10 10 10 10 10 
10 10 10 10 10 10 10 10 10 10 
 output

1 0 0 0 0 0 0 0 0 0 
0 1 0 0 0 0 0 0 0 0 
0 0 1 0 0 0 0 0 0 0 
0 0 0 1 0 0 0 0 0 0 
0 0 0 0 1 0 0 0 0 0 
0 0 0 0 0 1 0 0 0 0 
0 0 0 0 0 0 1 0 0 0 
0 0 0 0 0 0 0 1 0 0 
0 0 0 0 0 0 0 0 1 0 
0 0 0 0 0 0 0 0 0 1 
刚开始考虑不全面超时了

#include<iostream>
#include<cmath>
#include<cstring>
using namespace std;
int a[32][32],b[32][32],c[32][32];
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    memset(b,0,sizeof(b));
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++){
            scanf("%d",&a[i][j]);
            c[i][j]=a[i][j];
        }
    if(m==0)
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if(j!=0)
                    printf(" ");
                if(i==j)
                    printf("1");
                else
                    printf("0");
               }
            printf("\n");

            }
    else
    {
        m--;
        while(m--){
            for(int i=0;i<n;i++){
                for(int j=0;j<n;j++){
                    for(int k=0;k<n;k++){
                        b[i][j]+=a[i][k]*c[k][j];
                    }
                }
            }
            for(int i=0;i<n;i++)
                for(int j=0;j<n;j++)
                    c[i][j]=b[i][j];
            memset(b,0,sizeof(b));
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if(j!=0)
                    printf(" ");
                printf("%d",c[i][j]);

            }
            printf("\n");
        }

    }



    return 0;
}