洛谷CF799B T-shirt buying 题解

竟然又是一道STL的题！

好吧，我昨天刚发了一篇stl优先队列的题解。但是我觉得吧优先队列其时还是有很多可讲的，所以我单独出了一个优先队列的学习。

点这里啊！

好的优先队列就不讲了，直接先看题。

题目

一包共 n 件T恤被送到了商店。每件T恤被三个正整数 $p_{i}
​
a_{i}
​
和 b_{i}
​
所描述。 p_{i}
​
是第 i 件T恤的价格， a_{i}
​
是第 ii 件T恤正面的颜色， b_{i}
​
是第 ii 件T恤背面的颜色。所有 p_{i}
​
都各不相同， a_{i}
​
和 b_{i}
​
都是1~3的整数。
$

有 m 个顾客，每人都仅想买一件T恤。第 i 个顾客最喜欢的颜色是 c_{j}
​
。

如果一件T恤至少一面（前或后）的颜色是顾客喜欢的，他才会将其买下；如果有多件T恤符合条件，他会选择最便宜的；如果没有符合条件的，他什么都不会买。我们假定顾客都乖巧地排着队一个一个来，前一个顾客走后第二个才能被接待。

你需要计算每个顾客分别花了多少钱。

啊啊啊，好不清楚，还是看原题吧。

传送门

思路

首先用三个STL(标准模板库)里的优先队列存储每个颜色分别的价钱（需要用到结构体，要定义为从小到大（钱数）），然后使用哈希桶的思想创建bool类型数组用来存储此衣服有没有被买走（因为题目中说所有衣服价钱不一样）。接着输入输出就简了。

代码

定义结构体

struct node{
   
	int n,m;
	friend bool operator<(node a,node b)
	{
   
		return a.n>b.n;
	}
};

定义优先队列以及哈希表和一个数组方便存储

priority_queue <node> q1;
priority_queue <node> q2;
priority_queue <node> q3;
bool vis[2000005];
int a[2000005];

输入+进队

cin>>n;
	for(int i=1;i<=n;i++)
	{
   
		cin>>a[i];
	}
	for(int i=1;i<=n;i++)
	{
   
		int x;
		cin>>x;
		if(x==1)
		{
   
			q1.push(node{
   a[i],i});
		}
		if(x==2)
		{
   
			q2.push(node{
   a[i],i});
		}
		if(x==3)
		{
   
			q3.push(node{
   a[i],i});
		}
	}
	for(int i=1;i<=n;i++)
	{
   
		int x;
		cin>>x;
		if(x==1)
		{
   
			q1.push(node{
   a[i],i});