본문 바로가기
My Image
Algorithm/Programmers

[Algorithm] 행렬의 곱셈 (ProductMatrix)

by Lim-Ky 2017. 10. 4.
반응형

Programmers Level_02


행렬의 곱셈 (ProductMatrix)


행렬의 곱셈은, 곱하려는 두 행렬의 어떤 행과 열을 기준으로, 좌측의 행렬은 해당되는 행, 우측의 행렬은 해당되는 열을 순서대로 곱한 값을 더한 값이 들어갑니다. 행렬을 곱하기 위해선 좌측 행렬의 열의 개수와 우측 행렬의 행의 개수가 같아야 합니다. 곱할 수 있는 두 행렬 A,B가 주어질 때, 행렬을 곱한 값을 출력하는 productMatrix 함수를 완성해 보세요.



package Programmers_Level02;

class ProductMatrix {

	public static void main(String[] args) {
		ProductMatrix c = new ProductMatrix();
		int[][] a = { { 1, 2 , 3}, { 2, 3 ,4} };
		int[][] b = { { 3, 4 }, { 5, 6 },{ 3, 4 } };
      // 아래는 테스트로 출력해 보기 위한 코드입니다.
      System.out.println("행렬의 곱셈 : " + c.productMatrix(a, b));
      
      
	}


	public int[][] productMatrix(int[][] A, int[][] B) {
		int[][] answer = new int[A.length][B[0].length];
		
		for(int i=0; i<answer.length;i++){
			for(int j=0; j<answer[0].length;j++){
				
				for(int k=0; k<A[0].length;k++){
					answer[i][j] += A[i][k]*B[k][j];
				}
				
			}
			
		}
//		 for (int i = 0; i < answer.length; i++) {
//	            for (int j = 0; j < answer[i].length; j++) {
//	                System.out.print(answer[i][j] + " "); // 열 출력
//	            }
//	            System.out.println(); // 행 출력
//	        }
		
		return answer;
	}


}


반응형

댓글