有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
Python:
class Solution:
def timeRequiredToBuy(self, tickets: List[int], k: int) -> int:
total = 0
for i in range(len(tickets)):
if tickets[i] < tickets[k]:
total += tickets[i]
for i in range(k+1):
if tickets[i] >= tickets[k]:
total += tickets[k]
for j in range(k+1, len(tickets)):
if tickets[j] >= tickets[k]:
total += tickets[k]-1
return total
Java:
class Solution {
public int timeRequiredToBuy(int[] tickets, int k) {
int total = 0;
for(int a: tickets)
{
if(a < tickets[k])
total += a;
}
for(int i = 0; i <= k ; i++)
{
if(tickets[i] >= tickets[k])
total += tickets[k];
}
for(int i = k+1; i < tickets.length; i++)
{
if(tickets[i] >= tickets[k])
total += tickets[k]-1;
}
return total;
}
}
留言