Algorithm/Java

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

고고마코드 2021. 12. 20. 09:54
반응형

문제 링크

코딩테스트 연습 - 행렬의 덧셈 | 프로그래머스 (programmers.co.kr)


문제 이해하기

  • 행과 열이 같은 행렬의 덧셈

  • 같은 행, 같은 열의 값을 더한 결과

    • 행, 열 만큼 반복하며 같은 행, 열을 더하면 되겠다.

문제 풀이

코드-1

public int[][] solution2(int[][] arr1, int[][] arr2) {
    int[][] arr = new int[arr1.length][arr1[0].length];
    IntStream.range(0, arr1.length)
        .forEach(i -> IntStream.range(0, arr1[0].length)
            .forEach(j -> arr[i][j] = arr1[i][j] + arr2[i][j]));
    return arr;
}
  • 스트림을 활용해 람다식으로 풀어본 결과

코드-2

public int[][] solution(int[][] arr1, int[][] arr2) {
    int[][] arr = new int[arr1.length][arr1[0].length];
    for(int i=0; i<arr1.length; i++) {
        for(int j=0; j<arr1[0].length; j++) {
            arr[i][j] = arr1[i][j] + arr2[i][j];
        }
    }
    return arr;
}
  • 일반적인 반복문을 활용해 풀어본 결과

반응형