AtCoder Beginner Contest 170 E Smart Infants 堆+难点在于如何删除堆中指定节点

AtCoder Beginner Contest 170   比赛人数10527  比赛开始后4分钟看到A题,在比赛开始后第5分钟看到所有题

AtCoder Beginner Contest 170  E   Smart Infants   堆+难点在于如何删除堆中指定节点

总目录详见https://blog.csdn.net/mrcrack/article/details/104454762

在线测评地址https://atcoder.jp/contests/abc170/tasks/abc170_e

题目大意：具有不同等级的幼儿被安排在不同幼儿园，可以对幼儿所处的幼儿园进行变更，每进行一次变更，统计每个幼儿园中幼儿的最大等级，在不同幼儿园的最大等级中，寻找最小值，输出该最小值。

基本思路：看完题目，堆就跳了出来，难点在于如何删除堆中指定节点。

一个幼儿园开一个堆，再对所有幼儿园的堆顶开一个堆。

样例模拟如下：

6 3
8 1
6 2
9 3
1 1
2 2
1 3
4 3
2 1
1 2

6
2
6

安置好幼儿后的堆如下图哦所示。每个节点数据格式(幼儿编号，幼儿等级，幼儿存贮的幼儿园值，幼儿之前所处幼儿园值，幼儿现所处幼儿园)

第一次(4 3)变更后，对应的各堆如下

第二次(2 1)变更后，对应的各堆如下

第三次(1 2)变更后，对应的各堆如下

第一次全采用STL.编得挺辛苦，但收获也不小，总体感觉，该题的数据较弱。

AC代码如下

#include <cstdio>
#include <algorithm>
#include <vector>
#include <queue>
#define maxn 200010
using namespace std;
int N,Q,M;
vector<priority_queue<pair<int,int> > > q(maxn*3);
priority_queue<pair<int,int> > p;
vector<int> A(maxn),B(maxn);
int main(){
	int i,c,d,a,b;
	scanf("%d%d",&N,&Q);
	for(i=1;i<=N;i++)scanf("%d%d",&A[i],&B[i]),M=max(M,B[i]);//i对应幼儿序号
	for(i=1;i<=N;i++)//每个幼儿园建一个堆,此处i表示幼儿序号
		q[B[i]].push(make_pair(A[i],i));//节点信息：幼儿等级，幼儿序号
	for(i=1;i<=M;i++)//根据每个幼儿园的堆顶，再建一个堆,此处i表示幼儿园序号
		if(!q[i].empty())p.push(make_pair(-q[i].top().first,i));//节点信息:幼儿等级(技巧，采用负值)，幼儿园序号
	while(Q--){
		scanf("%d%d",&c,&d);
		b=B[c],B[c]=d;//b存储幼儿c之前所在幼儿园的信息;B[c]=d存储幼儿c现在所在幼儿园的信息。首先更新幼儿园堆的信息
		while(!q[b].empty()&&B[q[b].top().second]!=b)q[b].pop();//更新幼儿c之前所在幼儿园b的堆,保证堆顶幼儿就在幼儿园b
		if(!q[b].empty())p.push(make_pair(-q[b].top().first,b));
		q[d].push(make_pair(A[c],c));//更新幼儿c现在所在幼儿园B[c]=d的堆
		p.push(make_pair(-q[d].top().first,d));
		while(1){//更新每个幼儿园堆顶对应的堆的堆顶数据，保证其正确
			a=-p.top().first,b=p.top().second;//p的堆顶信息,请注意b对应的幼儿园信息，是未更新的信息
			if(q[b].empty())p.pop();//判定b幼儿园对应的堆是否有元素
			else if(q[b].top().first!=a)p.pop();//该幼儿在b幼儿园，但已不是q[b]的堆顶了
			else break;
		}
		printf("%d\n",-p.top().first);//注意再取一个负号,p堆顶元素，需幼儿等级，幼儿所属幼儿园与幼儿园堆顶元素一致时，才有效
	}
	return 0;
}