행렬의 덧셈

Java

class Solution {
    public int[][] solution(int[][] arr1, int[][] arr2) {
        int[][] answer = new int[arr1.length][];
        
        for(int i = 0; i < arr1.length; i++){
            answer[i] = new int[arr1[i].length];//각 행마다 열 초기화
            for(int j = 0; j < answer[i].length; j++){
                answer[i][j] = arr1[i][j] + arr2[i][j];
            }
        }
        return answer;
    }
}

Python

def solution(arr1, arr2):
    
    return [[x + y for x, y in zip(a, b)] for a, b in zip(arr1, arr2)]#2차원 배열이므로 2번