행렬의 덧셈

행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 2개의 행렬을 입력받는 sumMatrix 함수를 완성하여 행렬 덧셈의 결과를 반환해 주세요.

예를 들어 2x2 행렬인 A = ((1, 2), (2, 3)), B = ((3, 4), (5, 6)) 가 주어지면, 같은 2x2 행렬인 ((4, 6), (7, 9))를 반환하면 됩니다.(어떠한 행렬에도 대응하는 함수를 완성해주세요.)

단순한 행렬의 덧셈 문제인데 처음에 정사각행렬만 생각했다가 틀려서 다시 풀다가 A.length가 높이값이라는 것을 알게 되었다. 이전에는 전체 요소 크기인줄 알고 있었는데 바로 잡아서 기분이 좋다.

class SumMatrix {
    int[][] sumMatrix(int[][] A, int[][] B) {
        int[][] answer = new int[A.length][A[0].length];
        for(int i = 0; i < answer.length ; i++) {
            for(int j = 0; j < answer[0].length ; j++) {
                answer[i][j] = A[i][j] + B[i][j];
            }
        }
        
        return answer;
    }
 
    // 아래는 테스트로 출력해 보기 위한 코드입니다.
    public static void main(String[] args) {
        SumMatrix c = new SumMatrix();
        int[][] A = { { 1, 2 }, { 2, 3 } };
        int[][] B = { { 3, 4 }, { 5, 6 } };
        int[][] answer = c.sumMatrix(A, B);
        if (answer[0][0] == 4 && answer[0][1] == 6 && 
                answer[1][0] == 7 && answer[1][1] == 9) {
            System.out.println("맞았습니다. 제출을 눌러 보세요");
        } else {
            System.out.println("틀렸습니다. 수정하는게 좋겠어요");
        }
    }
}