每日一题 2582. 递枕头(简单)


思路

  1. 往返运动,要么在去的路上,要么在回来的路上
  2. 从 1 开始前进 n - 1 到达 n,从 n 开始前进 n - 1 到达 1,因此只需要判断 time // (n - 1) 的奇偶就可以知道传递方向
  3. 确定好方向后,使用取余操作就可以知道在该方向上前进了多少
class Solution:
    def passThePillow(self, n: int, time: int) -> int:
        return time % (n - 1) + 1 if (time // (n - 1) % 2 == 0) else n - time % (n - 1)

你可能感兴趣的:(用Python刷力扣,算法,leetcode,python)