每日一题|2073. 买票需要的时间|循环数组、下标分类

 

本题的数据范围比较大，如果是直接模拟的话会超时。

所以，一个更直接的想法是，在第k个人买完时，每个人分别都花了多少时间？

也就是，求出买票循环结束前每个人所花的时间。

那么，对于k之前的人，他们会在k买完票之前和k一样排到队首的机会，所以最多能够买ticket[k]的票，自身需要买ticket[i]，所以在二者之间取最小值。同时，k买的恰好是ticket[k]张票，所以两个可以合并成一个情况。when i <= k: time = min(ticket[k], ticket[i])。

对于k之后的人，他们在k买完之后那一轮无法排到队首，所以最多能买ticket[k] - 1张票。when i > k: time = min(ticket[i], ticket[k] - 1)。

从头到尾遍历全部的ticket，将每人的时间累积，得到最终结果。

class Solution(object):
    def timeRequiredToBuy(self, tickets, k):
        """
        :type tickets: List[int]
        :type k: int
        :rtype: int
        """
        res = 0
        n = len(tickets)

        for i in range(n):
            if i <= k:
                res += min(tickets[i], tickets[k])
            else:
                res += min(tickets[i], tickets[k] - 1)
        
        return res