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)]
풀이설명:
- 오른쪽으로 한 명을 건너뛰고 그다음 사람에게 던진다: 2칸씩 이동, 2를 곱해준다
- 0번 인덱스부터 시작해서 '공을 던지는 사람'을 찾아야 하므로 (k-1)
- % len(numbers): 3번째 예시에서처럼 넘어갈 수 있기에 % 연산자 사용
728x90
반응형
'코딩테스트 입문 > Python' 카테고리의 다른 글
[프로그래머스/programmers] 팩토리얼 (0) | 2023.07.19 |
---|---|
[프로그래머스/programmers] 최댓값 만들기(1) (0) | 2023.07.17 |
[프로그래머스/programmers] 합성수 찾기 (0) | 2023.07.17 |
[프로그래머스/programmers] 주사위의 개수 (0) | 2023.07.13 |
[프로그래머스/programmers] 배열 회전시키기 (2) | 2023.07.11 |