买票需要的时间

73 阅读2分钟

这是我参与「掘金日新计划 · 2 月更文挑战」的第 8 天,点击查看活动详情

问题描述

有 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

思路分析

首先我们先要理解一下题目意思,题目会给我们一个数组 tickets,代表一个买票的队伍,其中 tickets[i]表示第 i 个人需要购买 tickets[i]张票。购票时需要遵循以下规则:

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

  • 模拟排队计算

首先最直接的方法就是直接按照题目所说的方法进行模拟计算,这里题目的数据规模只有 100,所以直接模拟 O(n^2)的复杂度也是可以通过的。我们只需要重新遍历 tickets 数组,模拟购票,不断减少每个人所需购票的数量,直到第 k 个人购买完自己所需要的票数则结束循环。具体代码如下:

while (tickets[k] != 0)
  for (let i = 0; i < tickets.length; i++) {
    if (tickets[i] <= 0) continue;
    tickets[i]--;
    res++;
    if (tickets[k] === 0) break;
  }
  • 一次遍历快速计算

我们可以简单分析一下,第 k 个人所需要排队的次数应该就是其所需要购买的票数 tickets[k],那么他所需要等待的就是前面的人购买完 tickets[k]张票、以及后面的人购买完 tickets[k] - 1 张票的时间。但是前面的人有可能不需要购买 tickets[k]这么多张票,当他买完自己所需要的票数后就可以不用排队了,所以对于前面的人需要消耗的时间应该为 Math.min(tickets[i],tickets[k]);后面的人也是同理,为 Math.min(tickets[i], tickets[k] - 1)。具体代码如下:

for (let i = 0; i < tickets.length; i++) {
  if (i <= k) {
    res += Math.min(tickets[i], tickets[k]);
  } else if (i > k) {
    res += Math.min(tickets[i], tickets[k] - 1);
  }
}

完整 AC 代码如下:

AC 代码

/**
 * @param {number[]} tickets
 * @param {number} k
 * @return {number}
 */
var timeRequiredToBuy = function (tickets, k) {
  let res = 0;
  for (let i = 0; i < tickets.length; i++) {
    if (i <= k) {
      res += Math.min(tickets[i], tickets[k]);
    } else if (i > k) {
      res += Math.min(tickets[i], tickets[k] - 1);
    }
  }
  return res;
};

说在后面

本人为算法业余爱好者,平时只是随着兴趣偶尔刷刷题,如果上面分享有错误的地方,欢迎指出,感激不尽。