[Algorithm] 프로그래머스 - 공 던지기 (Kotlin)

친구들과 동그랗게 서서 공 던지기 게임을 한다. 공은 1번부터 던지며 오른쪽으로 한 명을 건너뛰고 그 다음 사람에게만 던질 수 있다. 친구들의 번호가 들어있는 정수 배열 numbers, 정수 k가 주어질 때 k번째로 공을 던지는 사람의 번호는 무엇인지 리턴하는 solution()을 완성하라

 

 

리스트를 복사하고 이어붙여서 풀어야 하는지도 고민해봤지만 그건 너무 오버하는 것 같아서 numbers 배열 안에서 노는 방법을 고민하다가 아래와 같이 코드를 작성했더니 성공했다.

 

class Solution {
    fun solution(numbers: IntArray, k: Int): Int = numbers[(2 * (k - 1) % numbers.size)]
}

 

numbers는 1부터 시작하기 때문에 k에서 1을 뺐고, numbers의 크기로 나눈 나머지를 인덱스로 해서 리턴하도록 짜니 테스트 케이스들을 통과했다.