[Lv. 0] 조건에 맞게 수열 변환하기 1

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/181882

프로그래머스

난이도 : Level.0

언어 : Kotlin

 

문제 설명 :

정수 배열 arr가 주어집니다. arr의 각 원소에 대해 값이 50보다 크거나 같은 짝수라면 2로 나누고, 50보다 작은 홀수라면 2를 곱합니다. 그 결과인 정수 배열을 return 하는 solution 함수를 완성해 주세요.

 

제한 사항 : 

• 1 ≤ arr의 길이 ≤ 1,000,000
  • 1 ≤ arr의 원소의 값 ≤ 100

입출력 예 :

arr	result
[1, 2, 3, 100, 99, 98]	[2, 2, 6, 50, 99, 49]

풀이 : 

class Solution {
    fun solution(arr: IntArray): IntArray = arr.map { if (it >= 50 && it % 2 == 0) it / 2 else if (it % 2 == 1) if (it * 2 >= 100) it else it * 2 else it }.toIntArray()
}