买票需要的时间
时间限制: 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;
}