首页 > 其他 > 详细

[BASIC-17] 矩阵乘法

时间:2014-05-22 08:18:30      阅读:411      评论:0      收藏:0      [点我收藏+]

基础练习 矩阵乘法  
时间限制: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
import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);

		while (scanner.hasNext()) {
			int n = scanner.nextInt();
			int m = scanner.nextInt();

			int[][] init = new int[n][n];
			for (int i = 0; i < n; i++) {
				for (int j = 0; j < n; j++) {
					init[i][j] = scanner.nextInt();
				}
			}

			int[][] result = new int[n][n];
			for (int i = 0; i < n; i++) {
				result[i][i] = 1;
			}

			powerOperation(result, init, m);
			printMatrix(result);
		}
	}

	private static void printMatrix(int[][] result) {
		int len = result.length;
		for (int i = 0; i < len; i++) {
			for (int j = 0; j < len; j++) {
				System.out.print(result[i][j]);
				System.out.print(j == len - 1 ? "\r\n" : " ");
			}
		}
	}

	private static void powerOperation(int[][] result, int[][] init, int m) {
		int len = init.length;
		for (int i = 0; i < m; i++) {		
			int[][] temp = new int[len][len];
			for (int j = 0; j < len; j++) {
				for (int j2 = 0; j2 < len; j2++) {
					for (int k = 0; k < len; k++) {
						temp[j][j2] += result[j][k] * init[k][j2];
					}
				}
			}

			for (int j = 0; j < len; j++) {
				for (int j2 = 0; j2 < len; j2++) {
					result[j][j2] = temp[j][j2];
				}
			}
		}
	}
}

[BASIC-17] 矩阵乘法,布布扣,bubuko.com

[BASIC-17] 矩阵乘法

原文:http://blog.csdn.net/u011506951/article/details/26238289

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!