基礎訓練-test17

問題描述

　　給定一個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

C語言程序：

#include<stdio.h>
#include<math.h>
int a[10][10]={0}, b[10][10]={0}, c[10][10]={0};

int main()
{
	int N, M, i, j, k, z, count;
	scanf("%d %d", &N,&M);
	if(N >= 1 && N <= 30 && M >= 0 && M <= 5)
	{
		for(i=0; i < N; i++)
		{
			for(j=0; j < N; j++)
			{
				scanf("%d",&a[i][j]);
				b[i][j] = a[i][j];
				if(abs(a[i][j]) > 10 || a[i][j] < 0)
					return 0;
			}
		}
		
		if(M==0)
		{
			for(i=0; i < N; i++)
			{
				c[i][i] =1;
			}
		}
		else if(M == 1)
		{
			for(i=0; i < N; i++)
				for(j=0; j < N; j++)
					c[i][j] = a[i][j];
		}
		else
		{
			for(count=1; count < M; count++)
			{
				for(i=0; i < N; i++)
				{
					for(j=0; j < N; j++)
					{
						c[i][j]=0;
						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++)
			{
				printf("%d ",c[i][j]);
			}
			printf("\n");
		}
	}
	return 0;
}