[Lv. 0] 마지막 두 원소

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

프로그래머스

난이도 : Level.0

언어 : Kotlin

 

문제 설명 :

정수 리스트 num_list가 주어질 때, 마지막 원소가 그전 원소보다 크면 마지막 원소에서 그전 원소를 뺀 값을 마지막 원소가 그전 원소보다 크지 않다면 마지막 원소를 두 배한 값을 추가하여 return하도록 solution 함수를 완성해주세요.

 

제한 사항 : 

• 2 ≤ num_list의 길이 ≤ 10
• 1 ≤ num_list의 원소 ≤ 9

입출력 예 :

num_list	result
[2, 1, 6]	[2, 1, 6, 5]
[5, 2, 1, 7, 5]	[5, 2, 1, 7, 5, 10]

풀이 : 

class Solution {
    fun solution(num_list: IntArray): IntArray = if (num_list[num_list.lastIndex] > num_list[num_list.lastIndex - 1]) num_list + (num_list[num_list.lastIndex] - num_list[num_list.lastIndex - 1]) else num_list + (num_list[num_list.lastIndex] * 2)
}