[프로그래머스][python] Lv.0 - 공던지기

📌 문제 설명

 

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

 

📌 작성한 코드

 

import math
def solution(numbers, k):
    answer = numbers[::2] if len(numbers)%2==0 else numbers[::2]+numbers[1::2]
    return answer[k%len(arr)-1]

 

else문에서 return 값을 numbers[1::2]만 넣어서 계속 원하는 답이 안나왔다 ㅜㅜ 생각은 해놓고 코드 짜면 잊어버리는 st ㅋㅋㅋㅋㅋㅋ 

그리고 원래 if-else문 썼다가 조건문 표현식으로 바꾸는 데 맛들린 것 같다 

코드도 훨씬 간편해진 넉끰 굿 👍👍