蓝桥杯BASIC-17 基础练习 矩阵乘法

蓝桥杯BASIC-17 基础练习 矩阵乘法

问题描述
　　给定一个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

  一开始交了好几次都是运行超时，我以为是循环的问题，简化了好几个循环之后还是不可以，没过的数据看了好几遍都以为是数组过大，就是没注意到0次幂是单位矩阵的问题，把仔细思考周全写在脑门上警示自己！！！！

上代码

#include <iostream>
#include <bits/stdc++.h>

using namespace std;


int main()
{
    int n, m;
    cin >> n >> m;
    int a[35][35];
    int b[35][35];
    int c[35][35];
    int i, j, k;
    for(i=0; i<n; i++){
        for(j=0; j<n; j++){
            cin >> a[i][j];
            b[i][j] = a[i][j];
        }
    }
    /*for(i=0; i<n; i++){
        for(j=0; j<n; j++)
            cout << b[i][j] << " ";
        cout << endl;
    }这地方是简化*/
    if(m == 0){
        for(i=0; i<n; i++){
            for(j=0; j<n; j++){
                if(i == j)
                    b[i][j] = 1;
                else
                    b[i][j] = 0;
            }
        }
    }
    else{
        int mm = m-1;
        while(mm--){
            /*for(i=0; i<n; i++){
                for(j=0; j<n; j++)
                    c[i][j] = 0;
            }下面简化，直接用memset初始化*/
            memset(c, 0, sizeof(c));
            for(i=0; i<n; i++){
                for(j=0; j<n; j++){
                    for(k=0; k<n; k++){
                        c[i][j] += b[i][k]*a[k][j];
                    }
                }
            }
            for(i=0; i<n; i++){
                for(j=0; j<n; j++)
                    b[i][j] = c[i][j];
            }
        }
    }
    for(i=0; i<n; i++){
        for(j=0; j<n; j++)
            cout << b[i][j] << " ";
        cout << endl;
    }
    return 0;
}