LeetCode 5926. 买票需要的时间

示例 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 秒。

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)