【数组】基础练习 矩阵乘法

基础练习 矩阵乘法

时间限制：1.0s 内存限制：512.0MB

问题描述

给定一个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;
long long int res[40][40]= {0};
int main()
{
	int n,m;
	long long int a[40][40],t[40][40];
	cin>>n>>m;
	for(int i=0; i<n; i++)
		for(int j=0; j<n; j++) {
			cin>>a[i][j];
			t[i][j]=a[i][j];
		}
//	每一行都去乘以每一列，得到结果该行每一列的值
	if(m==0) {
		for(int i=0; i<n; i++) {
			for(int j=0; j<n; j++) {
				if(i!=j)
					cout<<0<<" ";
				else
					cout<<1<<" ";
			}
			cout<<endl;
		}
		return 0;
	}
	while(--m) {
		for(int i=0; i<n; i++) {
			for(int j=0; j<n; j++) {
				res[i][j]=0;		
				int jrow=0,icol=0;
				while(jrow<n) {
					res[i][j]+=t[i][icol++]*a[jrow++][j];
				}
			}
		}
		for(int i=0; i<n; i++)
			for(int j=0; j<n; j++) {
				t[i][j]=res[i][j];				
			}
	}
	for(int i=0; i<n; i++) {
		for(int j=0; j<n; j++) {
			cout<<res[i][j]<<" ";
		}
		cout<<endl;
	}
}

柳神

#include <iostream>
using namespace std;
long long int b[40][40];
int main()
{
	int n, m;
	cin >> n >> m;
	long long int a[40][40];
	long long int t[40][40];
	for(int i = 0; i < n; i++) {
		for(int j = 0; j < n; j++) {
			cin >> a[i][j];
			t[i][j] = a[i][j];
		}
	}
	if(m == 0) {
		for(int i = 0; i < n; i++) {
			for(int j = 0; j < n; j++) {
				if(i != j) {
					cout << 0 << " ";
				} else {
					cout << 1 << " ";
				}
			}
			cout << endl;
		}
		return 0;
	}
	while(--m) {
		for(int i = 0; i < n; i++) {
			for(int j = 0; j < n; j++) {
				int k = n;
				while(k) {
					b[i][j] += t[i][k-1] * a[k-1][j];
					k--;
				}
			}
		}
		for(int i = 0; i < n; i++) {
			for(int j = 0; j < n; j++) {
				t[i][j] = b[i][j];
				b[i][j] = 0;
			}
		}
	}
	for(int i = 0; i < n; i++) {
		for(int j = 0; j < n; j++) {
			cout << t[i][j] << " ";
		}
		cout << endl;
	}
	return 0;
}

总结

习惯性想成res矩阵相乘。。。。。