LeetCode 5926. 买票需要的时间

·  阅读 131
LeetCode 5926. 买票需要的时间

题目

有 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

解题思路

class Solution:
    def timeRequiredToBuy(self, tickets: List[int], k: int) -> int:
        res = 0
        while True:
            for index,val in enumerate(tickets):
                # print(tickets,res)
                if val == 0:
                    continue
                else:
                    res += 1
                    tickets[index] -= 1
                    if index == k and tickets[index] == 0:
                        return res
        return res

if __name__ == '__main__':
    tickets = [2, 3, 2]
    k = 2
    res = Solution().timeRequiredToBuy(tickets, k)
    print(res)
分类:
代码人生
标签:
分类:
代码人生
标签:
收藏成功!
已添加到「」, 点击更改