[Python 걸음마] LeetCode 2582. Pass the Pillow 문제 풀이

문제

• https://leetcode.com/problems/pass-the-pillow/?envType=daily-question&envId=2024-07-06
• 입력: n, time
• 1번부터 n번까지의 사람이 일렬로 서있다. -> 1, 2, 3, ..., n
• 1번부터 시작해서 베개를 옆으로 토스, n번이 받으면 그 다음은 반대 방향으로 돌아감. -> n번이 (n-1)번에게 토스
• 출력: time만큼 토스했을 때, 베개는 몇 번째 사람이 가지고 있는가?

 

풀이 방법

• 처음부터 끝까지 토스되는데 걸리는 time은 (n-1)이다.
• 그렇다면 time을 (n-1)로 나눴을 때,
  • 몫이 짝수이면 정방향, 홀수이면 역방향임을 알 수 있다.
  • 그리고 그 나머지에 대해
    • 정방향이면 (나머지+1)번 째 사람이 베개를 가지게 되고,
    • 역방향이면 (n-나머지)번 째 사람이 베개를 가지게 된다.

 

코드

class Solution(object):
    def passThePillow(self, n, time):
        quotient, remainder = divmod(time, n-1)
        if quotient % 2 == 0:
            return remainder+1
        else:
            return n-remainder