본문 바로가기
Algorithm/Programmers

[Programmers] Lv.0 / 조건에 맞게 수열 변환하기 3 / Java

by unknownomad 2024. 3. 12.

문제

 

풀이

import java.util.Arrays;

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

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

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

 

출처

https://school.programmers.co.kr/learn/courses/30/lessons/181835?language=java

댓글