【东华大学oj】买票需要的时间

买票需要的时间

时间限制: 1s

类别: DS：队列->队列的应用

晚于 2024-05-26 23:59:00 后提交分数乘系数50%

截止日期：2024-05-30 23:59:00

问题描述

有 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和k，n表示tickets数组的长度，1 <= n <= 100；k表示位置，0 <= k < n

后n行输入n个整数，表示tickets数组的元素，1 <= 整数值val <= 100

输出说明

输出一个整数，表示结果

#include<iostream>
#include<stack>
#include<queue>
using namespace std;

int BuyTicket(int a[],int k,int n)
{
    a[k]=-1*a[k];//用负数来表示指定位置，因为队列一直会改动。
    queue<int>q;
    int time=0;
    for(int i=0; i<n; ++i)//入队
        q.push(a[i]);

    while(a[k]!=0)
    {
        time++;
        if(q.front()<0)//指定位置的票
        {
            q.front()++;
            a[k]=a[k]+1;
        }
        else q.front()--;//其余人所需票数减少

        if(q.front()!=0)
        {
            q.push(q.front());
            q.pop();
        }
        else q.pop();
    }
    return time;
}

int main()
{
    int n,k;
    cin>>n>>k;
    int a[1000];
    for(int i=0; i<n; ++i)
    {

        cin>>a[i];
    }
    cout<<BuyTicket(a,k,n)<<endl;
    return 0;
}