[프로그래머스] 행렬의 덧셈

이한솔·2023년 9월 27일
0

프로그래머스_레벨1

목록 보기
21/65
post-thumbnail

✨️ 문제 설명

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

제한 조건

  • 행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다

-> 자세한 내용 보러가기

🎲 자바 풀이

  • 첫번째 풀이
 class Solution {
    public int[][] solution(int[][] arr1, int[][] arr2) {
        int rows = arr1.length;
        int cols = arr1[0].length;
        int[][] answer = new int[rows][cols];
        for(int idx = 0; idx < rows; idx++){
            for(int j = 0; j < cols; j++){
                answer[idx][j] = arr1[idx][j] + arr2[idx][j];
            }
        }
        return answer;
    }
}
  • 두번째 풀이 : Stream 이용
 import java.util.stream.IntStream;

class Solution {
    public int[][] solution(int[][] arr1, int[][] arr2) {
        int rows = arr1.length;
        int cols = arr1[0].length;
        
        int[][] answer = new int[rows][cols];
        
        IntStream.range(0, rows)
            .forEach(idx -> IntStream.range(0, cols)
                .forEach(j -> answer[idx][j] = arr1[idx][j] + arr2[idx][j]));
        
        return answer;
    }
}

: 스트림을 이용해서도 풀 수 있지만, for문을 이용해서 푸는 게 이 경우 가독성이 좀 더 좋은 것 같다...

profile
개인 공부용

0개의 댓글