[LeetCode]5926. 买票需要的时间

买票需要的时间

Posted by JinFei on November 18, 2021

题目描述

有 n 个人前来排队买票,其中第 0 人站在队伍 最前方 ,第 (n - 1) 人站在队伍 最后方 。

给你一个下标从 0 开始的整数数组 tickets ,数组长度为 n ,其中第 i 人想要购买的票数为 tickets[i] 。

每个人买票都需要用掉 恰好 1 秒 。一个人 一次只能买一张票 ,如果需要购买更多票,他必须走到  队尾 重新排队(瞬间 发生,不计时间)。如果一个人没有剩下需要买的票,那他将会 离开 队伍。

返回位于位置 k(下标从 0 开始)的人完成买票需要的时间(以秒为单位)。

Example1:

输入:tickets = [2,3,2], k = 2 输出:6 解释:

  • 第一轮,队伍中的每个人都买到一张票,队伍变为 [1, 2, 1] 。
  • 第二轮,队伍中的每个都又都买到一张票,队伍变为 [0, 1, 0] 。 位置 2 的人成功买到 2 张票,用掉 3 + 3 = 6 秒。

Example2:

输入:tickets = [5,1,1,1], k = 0 输出:8 解释:

  • 第一轮,队伍中的每个人都买到一张票,队伍变为 [4, 0, 0, 0] 。
  • 接下来的 4 轮,只有位置 0 的人在买票。 位置 0 的人成功买到 5 张票,用掉 4 + 1 + 1 + 1 + 1 = 8 秒。

Note:

  • n == tickets.length
  • 1 <= n <= 100
  • 1 <= tickets[i] <= 100
  • 0 <= k < n

解题思路

class Solution {
public:
    int timeRequiredToBuy(vector<int>& tickets, int k) {
        unordered_map<int, int> mmap;
        int res = 0;
        queue<int> q;
        for(int i = 0; i < tickets.size(); i++){
            mmap[i] = tickets[i];
            q.push(i);
        }
        while(mmap[k] > 0){
            int id = q.front();
            q.pop();
            mmap[id]--;
            if(mmap[id] > 0){
                q.push(id);
            }
            res++;
        }
        return res;
    }
};