【CF799B】T-shirt buying（一道很水的小根堆）

点此看题面

大致题意： 有\(n\)件T恤衫，告诉你每件T恤衫的价格以及它正面和反面的颜色（\(1≤\)颜色的编号\(≤3\)），现在有m个顾客，已知每个人想要的衣服的颜色（一件T恤衫只要有一面的颜色满足他的需求即可），请你求出每个人所需支付的最低价格（一件T恤衫只能被买一次）。

题解

由于这道题的颜色最多只有\(3\)种，因此，我们只要开\(3\)个小根堆来存储每种颜色的衣服的价格和编号（价格为第一关键字）即可。

由于每件衣服只能被买一次，因此当一件衣服被买走之后，我们可以给它打个标记表示它已被买走，下次再遇到它时就可以直接跳过它了。

代码

#include<bits/stdc++.h>

#define max(x,y) ((x)>(y)?(x):(y))

#define min(x,y) ((x)<(y)?(x):(y))

#define N 200000

using namespace std;

int n,Q,del[N+5];

struct clothes

{

	int a,b,pos,Price;

	bool operator < (const clothes a) const

	{

		return Price>a.Price;

	}

}s[N+5];

priority_queue<clothes> h[4];

inline char tc()

{

	static char ff[100000],*A=ff,*B=ff;

	return A==B&&(B=(A=ff)+fread(ff,1,100000,stdin),A==B)?EOF:*A++;

}

inline void read(int &x)

{

	x=0;char ch;

	while(!isdigit(ch=tc()));

	while(x=(x<<3)+(x<<1)+ch-'0',isdigit(ch=tc()));

}

inline void write(int x)

{

	if(x>9) write(x/10);

	putchar(x%10+'0');

}

int main()

{

	register int i;

	for(read(n),i=1;i<=n;++i) read(s[i].Price);

	for(i=1;i<=n;++i) read(s[i].a),s[i].pos=i;

	for(i=1;i<=n;++i) read(s[i].b),h[s[i].a].push(s[i]),h[s[i].b].push(s[i]);

	for(read(Q);Q;--Q)

	{

		int x;read(x);

		while(!h[x].empty()&&del[h[x].top().pos]) h[x].pop();//若当前的衣服已被买走，则直接跳过它

		if(h[x].empty()) putchar('-'),putchar('1'),putchar(' ');//如果堆已空，说明没有这种颜色的衣服了，输出-1

		else write(h[x].top().Price),putchar(' '),del[h[x].top().pos]=1,h[x].pop();//输出答案，并标记这件衣服已被买走

	}

	return 0;

}