[JAVA 문제 풀이] 207. 조건에 맞게 수열 변환하기 3

내가 작성한 코드 1 (반복문)

class Solution {
    public int[] solution(int[] arr, int k) {
        if(k % 2 == 0) {
            for (int i = 0; i < arr.length; i++) {
                arr[i] += k;
            }
        } else {
            for (int i = 0; i < arr.length; i++) {
                arr[i] *= k;
            }
        }

        return arr;
    }
}

내가 작성한 코드 2 (스트림)

import java.util.*;

class Solution {
    public int[] solution(int[] arr, int k) {
        return Arrays.stream(arr).map(num -> k % 2 == 0 ? num + k : num * k).toArray();
    }
}