[알고리즘문제풀기] 조건에 맞게 수열 변환하기 3

문제

내가 작성한 정답

for

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

Arrays.stream

import java.util.Arrays;

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

IntStream

import java.util.stream.IntStream;

class Solution {
    public int[] solution(int[] arr, int k) {
        return IntStream.range(0,arr.length)
                .map(i->(k%2==1)?arr[i]*k:arr[i]+k)
                .toArray();

    }
}