【洛谷 P2827】蚯蚓【堆—＞队列】

---

解题思路

对于每一秒除了被切的哪一个所有的蚯蚓都增长Q米，我们来维护3个队列，队列1表示最开始的蚯蚓，队列2表示每一次被切的蚯蚓被分开的较长的那一部分，队列3表示每一次被切的蚯蚓被分开的较短的那一部分。
我们先把原序列排序，因为不管怎么切，先被切的蚯蚓分成的两部分一定比后切的蚯蚓分成的两部分大(较大的部分和较大的部分比较，较小的部分和较小的部分比较)，所以我们可以省去每一秒增加每只蚯蚓的长度这个操作，转换成在查询砍那只蚯蚓时，把增加的长度算到蚯蚓的总长度上

寻找每次砍哪一只蚯蚓就是在队列1、队列2、队列3的队头找一个算上增加的长度最大的蚯蚓，之后把他出队，切开的两部分分别进入队2、队3。

对于增量的计算我们可以按照蚯蚓在队列中的标号，因为队列1中的蚯蚓直到被切是一直处于一种增长状态，所以直接加上(当前时间-1) * Q就可以了，而对于队列2和队列3的蚯蚓，他的增长是从被切掉那一刻的下一秒开始的，所以他的增长量则是(当前时间-1-被切割的时间)*Q。

---

代码

#include<cstdio>
#include<iostream>
#include<queue>
#include<algorithm>
#include<cstring>
#include<cmath>
#define ll long long
using namespace std;

int n,m,q,u,v,t,y,ok,nn;
ll ans[7500010],a[7500010];
double p;

queue<int>q1,q2,q3;

ll get_max()
{
	ll maxn=-(1<<30);
	if(!q1.empty()&&q1.front()>maxn)ok=1,maxn=q1.front();
	if(!q2.empty()&&q2.front()>maxn)ok=2,maxn=q2.front();
	if(!q3.empty()&&q3.front()>maxn)ok=3,maxn=q3.front();
	if(ok==1)q1.pop();
	if(ok==2)q2.pop();
	if(ok==3)q3.pop();
	return maxn;
}

bool cmp(int x,int y)
{
	return x>y;
}

int main(){
	scanf("%d%d%d%d%d%d",&n,&m,&q,&u,&v,&t);
	for(int i=1;i<=n;i++)
		scanf("%lld",&a[i]);
	sort(a+1,a+n+1,cmp);
	for(int i=1;i<=n;i++)
		q1.push(a[i]);
	for(int i=1;i<=m;i++)
	{
		ll x=get_max()+y;
		ans[i]=x;
		ll x1=x*u/v*1.0,x2=x-x1;
		y+=q;
		if(x1>x2)swap(x1,x2);
		q2.push(x1-y),q3.push(x2-y);
	}
	for(int i=t;i<=m;i+=t)
		printf("%lld ",ans[i]);
	cout<<endl;
	while(!q1.empty()||!q2.empty()||!q3.empty())ans[++nn]=get_max()+y;
	for(int i=t;i<=nn;i+=t)
		printf("%lld ",ans[i]);
}