코딩 테스트 - 조건에 맞게 수열 변환하기 3

package programmers;

import java.util.stream.IntStream;

class 조건에_맞게_수열_변환하기_3_Java_230808 {
    /*
    문제 설명
        정수 배열 arr와 자연수 k가 주어집니다.

        만약 k가 홀수라면 arr의 모든 원소에 k를 곱하고, k가 짝수라면 arr의 모든 원소에 k를 더합니다.

        이러한 변환을 마친 후의 arr를 return 하는 solution 함수를 완성해 주세요.

    제한사항
        1 ≤ arr의 길이 ≤ 1,000,000
        1 ≤ arr의 원소의 값 ≤ 100
        1 ≤ k ≤ 100

    입출력 예
        arr    k  result
        [1, 2, 3, 100, 99, 98] 3  [3, 6, 9, 300, 297, 294]
        [1, 2, 3, 100, 99, 98] 2  [3, 4, 5, 102, 101, 100]

    입출력 예 설명
        입출력 예 #1

        주어진 k인 3은 홀수이므로, 전체 배열에 3을 곱합니다. 따라서 [3, 6, 9, 300, 297, 294]을 return 합니다.
        입출력 예 #2

        주어진 k인 2는 짝수이므로, 전체 배열에 2를 더합니다. 따라서 [3, 4, 5, 102, 101, 100]을 return 합니다.
     */


    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();
    }
    public int[] solution2(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;
    }
}

package programmers

class 조건에_맞게_수열_변환하기_3_Kotlin_230808 {
    fun solution(arr: IntArray, k: Int): IntArray {
        return if (k % 2 == 0) {
            arr.map { it + k }.toIntArray()
        } else {
            arr.map { it * k }.toIntArray()
        }
    }

    fun solution2(arr: IntArray, k: Int): IntArray {
        if (k % 2 == 0) {
            for (i in arr.indices) {
                arr[i] += k
            }
        } else {
            for (i in arr.indices) {
                arr[i] *= k
            }
        }
        return arr
    }


    fun solution_short(arr: IntArray, k: Int) = arr.map { if (k % 2 == 0) it + k else it * k }

}