九度题目1474 矩阵幂 北邮计算机机试2012

时间:2022-05-03 08:55:18

这道题一开始忽略了一次幂的情况,WA了两次,重新看题恍然大悟,果然审题需要仔细。。


#include <stdio.h>
struct Matrix{
    int size;
    int times;
    int a[10][10];
    int temp[10][10];
    int b[10][10];
}matrix[10];//对于多组数据,需要保存结果的,尽量用结构体
void mul(int (*temp)[10],int (*a)[10],int (*b)[10],int size)
{
    for(int k=0;k<size;k++)
    {
        for(int i=0;i<size;i++)
        {
            for(int j=0;j<size;j++)
            {
                temp[k][i]+=b[k][j]*a[j][i];
            }
        }
    }
}
int main()
{
    int n=0;
    while(~scanf("%d",&n))
    {
        for(int index=0;index<n;index++)
        {
            scanf("%d %d",&matrix[index].size,&matrix[index].times);
            for(int i=0;i<matrix[index].size;i++)
            {
                for(int j=0;j<matrix[index].size;j++)
                {
                    scanf("%d",&matrix[index].a[i][j]);
                    matrix[index].b[i][j]=matrix[index].a[i][j];
                }
            }
            if(matrix[index].times==1)//如果是1次幂,对matrix[index].temp[i][j]赋值,然后进行下一次循环
            {
                for(int i=0;i<matrix[index].size;i++)
                {
                    for(int j=0;j<matrix[index].size;j++)
                    {
                        matrix[index].temp[i][j]=matrix[index].a[i][j];
                    }
                }
                continue;
            }
            for(int i=2;i<=matrix[index].times;i++)
            {
                for(int i=0;i<matrix[index].size;i++)
                {
                    for(int j=0;j<matrix[index].size;j++)
                    {
                        matrix[index].temp[i][j]=0;
                    }
                }
                mul(matrix[index].temp,matrix[index].a,matrix[index].b,matrix[index].size);
                for(int i=0;i<matrix[index].size;i++)
                {
                    for(int j=0;j<matrix[index].size;j++)
                    {
                        matrix[index].b[i][j]=matrix[index].temp[i][j];
                    }
                }
            }
        }
        for(int index=0;index<n;index++)
        {
            for(int i=0;i<matrix[index].size;i++)
            {
                for(int j=0;j<matrix[index].size;j++)
                {
                    if(j==matrix[index].size-1)
                    {
                        printf("%d\n",matrix[index].temp[i][j]);
                    }
                    else
                    {
                        printf("%d ",matrix[index].temp[i][j]);
                    }
                }
            }
        }       
    }
    return 0;
}
/**************************************************************
    Problem: 1474
    User: 爱理momoko
    Language: C
    Result: Accepted
    Time:0 ms
    Memory:924 kb
****************************************************************/