프로그래머스: 행렬의 덧셈

최창효·2022년 1월 15일
0
post-thumbnail

문제 설명

  • 행렬의 덧셈 연산을 실행하는 문제입니다.

정답

파이썬

def solution(arr1, arr2):
    answer = []
    for a,b in zip(arr1,arr2):
        col = []
        for q,w in zip(a,b):
            col.append(q+w)
        answer.append(col)
    return answer

자바

class Solution {
    public int[][] solution(int[][] arr1, int[][] arr2) {
    	// 행의 길이가 arr1.length이고 열의 길이가 arr1[0].length이며 정수값을 담는 배열 answer를 선언합니다
        int[][] answer = new int [arr1.length][arr1[0].length];
        for (int i=0;i<arr1.length;i++){ // 행 
            for (int j=0;j<arr1[i].length;j++){ // 열
                    answer[i][j] = arr1[i][j]+arr2[i][j]; // arr1과 arr2의 값을 더합니다
            }
        }
        return answer;
    }
}

기타

  • 자바 배열의 선언초기화를 잘 기억해두자
  • 자바에는 파이썬의 zip함수와 동일한 거 없나?
profile
기록하고 정리하는 걸 좋아하는 개발자.

0개의 댓글