买票需要的时间

标签: 队列 数组 模拟

难度: Easy

有 n 个人前来排队买票，其中第 0 人站在队伍 最前方 ，第 (n - 1) 人站在队伍 最后方 。

给你一个下标从 0 开始的整数数组 tickets ，数组长度为 n ，其中第 i 人想要购买的票数为 tickets[i] 。

每个人买票都需要用掉 恰好 1 秒 。一个人 一次只能买一张票 ，如果需要购买更多票，他必须走到  队尾 重新排队（瞬间 发生，不计时间）。如果一个人没有剩下需要买的票，那他将会 离开 队伍。

返回位于位置 k（下标从 0 开始）的人完成买票需要的时间（以秒为单位）。

示例 1：

输入：tickets = [2,3,2], k = 2
输出：6
解释： 
- 第一轮，队伍中的每个人都买到一张票，队伍变为 [1, 2, 1] 。
- 第二轮，队伍中的每个都又都买到一张票，队伍变为 [0, 1, 0] 。
位置 2 的人成功买到 2 张票，用掉 3 + 3 = 6 秒。

示例 2：

输入：tickets = [5,1,1,1], k = 0
输出：8
解释：
- 第一轮，队伍中的每个人都买到一张票，队伍变为 [4, 0, 0, 0] 。
- 接下来的 4 轮，只有位置 0 的人在买票。
位置 0 的人成功买到 5 张票，用掉 4 + 1 + 1 + 1 + 1 = 8 秒。

提示：

  • n == tickets.length
  • 1 <= n <= 100
  • 1 <= tickets[i] <= 100
  • 0 <= k < n

Submission

运行时间: 24 ms

内存: 16.0 MB

class Solution:
    def timeRequiredToBuy(self, tickets: List[int], k: int) -> int:
        n = len(tickets)
        sumTime = 0
        for i in range(n):
            if i <= k:
                sumTime += min(tickets[i],tickets[k])
            else:
                sumTime += min(tickets[i],tickets[k]-1)

        return sumTime

Explain

这个题解采用了一种简单直观的方法。对于队列中的每一个人，我们需要计算他们买票所需的时间。如果这个人在k位置之前（包括k位置），那么他们将买到tickets[k]张票，因为当轮到k位置的人时，队列前面的人都会买到相同数量的票。如果这个人在k位置之后，那么他们将买到tickets[k]-1张票，因为当轮到k位置的人买完票后，这个人还需要再排队一次。

时间复杂度: O(n)

空间复杂度: O(1)

class Solution:
    def timeRequiredToBuy(self, tickets: List[int], k: int) -> int:
        n = len(tickets)  # 获取队列长度
        sumTime = 0  # 初始化总时间
        for i in range(n):
            if i <= k:
                sumTime += min(tickets[i], tickets[k])  # 如果在k之前或等于k，买票时间取决于自身和k位置的较小值
            else:
                sumTime += min(tickets[i], tickets[k] - 1)  # 如果在k之后，买票时间取决于自身和k位置减1的较小值
        return sumTime  # 返回总时间