공 던지기

https://school.programmers.co.kr/learn/courses/30/lessons/120843

프로그래머스

 

[문제 설명]

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

 

제한사항

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

 

[나의 풀이]

numbers의 원소의 개수가 짝수개이면 홀수번호만 반복을 하고(ex.135135), 홀수개이면 홀수번호 뭉텅이와 짝수 번호뭉텅이가 번갈아가면서 나온다고(ex.132132) 생각해서 경우를 나눠서 코드를 작성하였다.

def solution(numbers, k):
    a = []
    b = []
    if len(numbers) % 2 == 0:
        for i in range(1,len(numbers),2):
            a.append(i)
        return a[k %len(a)-1] 
                    
    else:
        for i in range(1,len(numbers)+1,2):
            a.append(i)
        for i in range(2,len(numbers),2):
            b.append(i)
        c = a + b
        return c[k%len(c)-1]

 

그런데 챗지피티가 훨씬 간단하게 코드를 짜줬다^^..

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

 

[다른 풀이]

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

def solution(numbers, k):
    idx = 0
    for i in range(k-1):
        idx += 2
        idx %= len(numbers)

    return idx + 1

from collections import deque
def solution(numbers, k):
    array = deque(numbers)

    array.rotate(-(k-1)*2)

    return array[0]