LeetCode 2073.买票需要的时间：硬卷O(n)——一次遍历

【LetMeFly】2073.买票需要的时间：硬卷O(n)——一次遍历

力扣题目链接：https://leetcode.cn/problems/time-needed-to-buy-tickets/

有 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

解题方法：一次遍历

试想，下标k买完所有票后，其他人分别买了几张票？

• 如果这个人i在k及其之前，那么他最多买tickets[k]张票，他实际买了min(tickets[i], tickets[k])张。
• 如果这个人i在k及其之后，那么他最多买tickets[k] - 1张票，他实际买了min(tickets[i], tickets[k] - 1)张。

因此，一次遍历即可得到答案了。

• 时间复杂度$O(n)$
• 空间复杂度$O(1)$

AC代码

C++

class Solution {
public:
    int timeRequiredToBuy(vector<int>& tickets, int k) {
        int ans = 0;
        for (int i = 0; i < tickets.size(); i++) {
            ans += i <= k ? min(tickets[i], tickets[k]) : min(tickets[i], tickets[k] - 1);
        }
        return ans;
    }
};

Go

package main

func timeRequiredToBuy(tickets []int, k int) int {
    ans := 0
    for th, t := range tickets {
        if th <= k {
            ans += min(t, tickets[k])
        } else {
            ans += min(t, tickets[k] - 1)
        }
    }
    return ans
}

执行用时分布0ms击败100.00%复杂度分析；消耗内存分布2.10MB击败100.00%。

Java

class Solution {
    public int timeRequiredToBuy(int[] tickets, int k) {
        int ans = 0;
        for (int i = 0; i < tickets.length; i++) {
            if (i <= k) {
                ans += Math.min(tickets[i], tickets[k]);
            } else {
                ans += Math.min(tickets[i], tickets[k] - 1);
            }
        }
        return ans;
    }
}

Python

from typing import List

class Solution:
    def timeRequiredToBuy(self, tickets: List[int], k: int) -> int:
        # return sum(min(t, tickets[k] if th <= k else tickets[k] - 1) for th, t in enumerate(tickets))
        return sum(min(t, tickets[k] - (th > k)) for th, t in enumerate(tickets))

同步发文于CSDN和我的个人博客，原创不易，转载经作者同意后请附上原文链接哦~

Tisfy：https://letmefly.blog.csdn.net/article/details/142691600