九度題目1474 矩陣冪 北郵計算機機試2012

這道題一開始忽略了一次冪的情況,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
****************************************************************/


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章