题目:
有 n
个人前来排队买票,其中第 0
人站在队伍 最前方 ,第 (n - 1)
人站在队伍 最后方 。
给你一个下标从 0 开始的整数数组 tickets
,数组长度为 n
,其中第 i
人想要购买的票数为 tickets[i]
。
每个人买票都需要用掉 恰好 1 秒 。一个人 一次只能买一张票 ,如果需要购买更多票,他必须走到 队尾 重新排队(瞬间 发生,不计时间)。如果一个人没有剩下需要买的票,那他将会 离开 队伍。
返回位于位置 k
(下标从 0 开始)的人完成买票需要的时间(以秒为单位)。
示例 1:
输入:tickets = [2,3,2], k = 2
输出:6
解释:
- 队伍一开始为 [2,3,2],第 k 个人以下划线标识。
- 在最前面的人买完票后,队伍在第 1 秒变成 [3,2,1]。
- 继续这个过程,队伍在第 2 秒变为[2,1,2]。
- 继续这个过程,队伍在第 3 秒变为[1,2,1]。
- 继续这个过程,队伍在第 4 秒变为[2,1]。
- 继续这个过程,队伍在第 5 秒变为[1,1]。
- 继续这个过程,队伍在第 6 秒变为[1]。第 k 个人完成买票,所以返回 6。
示例 2:
输入:tickets = [5,1,1,1], k = 0
输出:8
解释:
- 队伍一开始为 [5,1,1,1],第 k 个人以下划线标识。
- 在最前面的人买完票后,队伍在第 1 秒变成 [1,1,1,4]。
- 继续这个过程 3 秒,队伍在第 4 秒变为[4]。
- 继续这个过程 4 秒,队伍在第 8 秒变为[]。第 k 个人完成买票,所以返回 8。
提示:
n == tickets.length
1 <= n <= 100
1 <= tickets[i] <= 100
0 <= k < n
思路:
直接通过队列模拟买票情况即可
当某个人买完票后,他不再入队
代码:
class Solution {
public:
int timeRequiredToBuy(vector<int>& tickets, int k) {
int n = tickets.size();
int ans = 0;
queue<pair<int, int>> que;
for (int i = 0; i < tickets.size(); i++) {
que.push(make_pair(tickets[i], i));
}
while (true) {
auto top = que.front();
if (top.first == 1 && top.second == k) {
return ++ans;
}
que.pop();
ans++;
if (top.first == 1) continue;
que.push(make_pair(--top.first,top.second));
}
return -1;
}
};