基础练习 矩阵乘法

基础练习 矩阵乘法
时间限制：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

#include<iostream>
using namespace std;

int main()
{
    int d[32][32]={0},l[32][32]={0},f[32][32]={0},n,m,index;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++){
            cin>>d[i][j];
            l[i][j]=d[i][j];
        }
    }
    if(m!=0)
    {
    while(--m)
    {
        for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
                index=0;
            for(int k = 1;k<=n;k++){
                    index+=l[i][k]*d[k][j];
                }
        f[i][j]=index;
            }
        }
        for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
                l[i][j]=f[i][j];
        }
        }
    }

    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<n;j++){
            cout<<f[i][j]<<" ";
        }
        cout<<f[i][n]<<endl;
    }
    }else{
        for(int i=1;i<=n;i++)
    {
        for(int j=1;j<n;j++){
            if(i!=j)cout<<f[i][j]<<" ";
            else cout<<"1 ";
        }
        if(i!=n)cout<<f[i][n]<<endl;
            else cout<<"1"<<endl;
    }
    }

    return 0;
}