본문 바로가기
코딩테스트 입문/Python

[프로그래머스/programmers] 공 던지기

by Peter.JH 2023. 7. 10.
728x90

문제:

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

 

 

입력:

 친구들의 번호가 들어있는 정수 배열 numbers와 정수 K

 

출력:

 k번째로 공을 던지는 사람의 번호

 

 

제한사항:

  • 2 < numbers의 길이 < 100
  • 0 < k < 1,000
  • numbers의 첫 번째와 마지막 번호는 실제로 바로 옆에 있습니다.
  • numbers는 1부터 시작하며 번호는 순서대로 올라갑니다.
  • 공은 1번부터 던지며 오른쪽으로 한 명을 건너뛰고 그다음 사람에게만 던질 수 있습니다.

 

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

 

solution.py:

 

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

 

풀이설명: 

 

  1. 오른쪽으로 한 명을 건너뛰고 그다음 사람에게 던진다: 2칸씩 이동, 2를 곱해준다
  2.  0번 인덱스부터 시작해서 '공을 던지는 사람'을 찾아야 하므로 (k-1)
  3. % len(numbers): 3번째 예시에서처럼 넘어갈 수 있기에 % 연산자 사용

 

728x90