공 던지기

송용진·2023년 9월 9일
0

알고리즘

목록 보기
115/173

머쓱이는 친구들과 동그랗게 서서
공 던지기 게임을 하고 있습니다.

공은 1번부터 던지며
오른쪽으로 한 명을 건너뛰고
그다음 사람에게만 던질 수 있습니다.

친구들의 번호가 들어있는 정수 배열 numbers와
정수 K가 주어질 때,
k번째로 공을 던지는 사람의 번호는 무엇인지 return 하도록 solution 함수를 완성해보세요.

내 코드

def solution(numbers, k):
    answer = 0
    i = 0
    while k:                
        if i > len(numbers)-1:
            i = i - len(numbers)            
        answer = numbers[i]
        i += 2
        k -= 1
    return answer

예시 코드

def solution(numbers, k):
    return numbers[2 * (k - 1) % len(numbers)]
profile
백엔드 개발자

0개의 댓글