[프로그래머스:PYTHON] 공던지기

문제링크 :

프로그래머스

문제 :

 

공던지기

---

문제 설명

머쓱이는 친구들과 동그랗게 서서 공 던지기 게임을 하고 있습니다. 공은 1번부터 던지며 오른쪽으로 한 명을 건너뛰고 그다음 사람에게만 던질 수 있습니다. 친구들의 번호가 들어있는 정수 배열 numbers와 정수 K가 주어질 때, k번째로 공을 던지는 사람의 번호는 무엇인지 return 하도록 solution 함수를 완성해보세요.

 

제한사항

• 2 < numbers의 길이 < 100
• 0 < k < 1,000
• numbers의 첫 번째와 마지막 번호는 실제로 바로 옆에 있습니다.
• numbers는 1부터 시작하며 번호는 순서대로 올라갑니다.

 

입출력 예

numbers	k	result
[1,2,3,4]	2	3
[1, 2, 3, 4, 5, 6]	5	3
[1, 2, 3]	3	2

 

입출력 예 설명 1

• 1번은 첫 번째로 3번에게 공을 던집니다.
• 3번은 두 번째로 1번에게 공을 던집니다.

 

입출력 예 설명 2

• 1번은 첫 번째로 3번에게 공을 던집니다.
• 3번은 두 번째로 5번에게 공을 던집니다.
• 5번은 세 번째로 1번에게 공을 던집니다.
• 1번은 네 번째로 3번에게 공을 던집니다.
• 3번은 다섯 번째로 5번에게 공을 던집니다.

입출력 예 설명 3

• 1번은 첫 번째로 3번에게 공을 던집니다.
• 3번은 두 번째로 2번에게 공을 던집니다.
• 2번은 세 번째로 1번에게 공을 던집니다.

 

풀이과정

문제풀이 로직

1. 2개씩 건너 뛰었을 때, k번째 공을 던지는 사람을 찾으면 2*(k-1)을 해주어야 한다. 2*k를 하면 공을 받는 사람을 찾아야 하기 때문!!

 2. 2*k(k-1)이 numbers의 길이를 초과하는 경우, 리스트를 확장하는 대신 %를 사용하여 해당되는 인덱스 구하기. 

 

내 풀이

def solution(numbers, k):
    return numbers[2 * (k - 1) % len(numbers)]