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});
		}
		if(x==2)
		{
			q2.push(node{a[i],i});
		}
		if(x==3)
		{
			q3.push(node{a[i],i});
		}
	}

判断输出

cin>>m;
	for(int i=1;i<=m;i++)
	{
		int y;
		cin>>y;
		if(y==1)
		{
			while(true)
			{
				if(q1.empty()){
					cout<<-1<<" ";
					break;
				}
				if(!vis[q1.top().m])
				{
					cout<<q1.top().n<<" ";
					vis[q1.top().m]=true;
					q1.pop();
					break;
				}
				else q1.pop();
			}
		}
		if(y==2)
		{
			while(true)
			{
				if(q2.empty()){
					cout<<-1<<" ";
					break;
				}
				if(!vis[q2.top().m])
				{
					cout<<q2.top().n<<" ";
					vis[q2.top().m]=true;
					q2.pop();
					break;
				}
				else q2.pop();
			}
		}
		if(y==3)
		{
			while(true)
			{
				if(q3.empty()){
					cout<<-1<<" ";
					break;
				}
				if(!vis[q3.top().m])
				{
					cout<<q3.top().n<<" ";
					vis[q3.top().m]=true;
					q3.pop();
					break;
				}
				else q3.pop();
			}
		}
	}

完整代码+注释

#include<iostream>
#include<queue>//优先队列头文件 
using namespace std;
struct node{//结构体 
	int n,m;
	friend bool operator<(node a,node b)//朋友函数 
	{
		return a.n>b.n;//小顶堆 
	}
};
priority_queue <node> q1;//颜色1优先队列 
priority_queue <node> q2;//颜色2优先队列 
priority_queue <node> q3;//颜色3优先队列 
bool vis[2000005];//哈希表 
int a[2000005];//数组 
int main()
{
	int n,m;
	cin>>n;//衣服个数 
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];//输入钱数进a 
	}
	for(int i=1;i<=n;i++)//正面 
	{
		int x;
		cin>>x;
		if(x==1)//颜色1 
		{
			q1.push(node{a[i],i});//让编号和钱数进q1 
		}
		if(x==2)
		{
			q2.push(node{a[i],i});//让编号和钱数进q2 
		}
		if(x==3)
		{
			q3.push(node{a[i],i});//让编号和钱数进q3 
		}
	}
	for(int i=1;i<=n;i++)//背面 
	{
		int x;
		cin>>x;
		if(x==1)//颜色1 
		{
			q1.push(node{a[i],i});//同上 
		}
		if(x==2)//颜色2 
		{
			q2.push(node{a[i],i});//同上 
		}
		if(x==3)//颜色3 
		{
			q3.push(node{a[i],i});//同上 
		}
	}
	cin>>m;//人数 
	for(int i=1;i<=m;i++)
	{
		int y;
		cin>>y;//颜色 
		if(y==1)//1
		{
			while(true)
			{
				if(q1.empty()){//如果对空还没有合适 
					cout<<-1<<" ";//对不起，货没了 
					break;
				}
				//哈希表中可买，且符合全部要求的最小钱数 
				if(!vis[q1.top().m])//找到合适的了 
				{
					cout<<q1.top().n<<" ";//输出队顶原来编号 
					vis[q1.top().m]=true;//不可以再买了 
					q1.pop();//删除 
					break;//结束循环 
				}
				else q1.pop();//否则下一个 
			}
		}
		if(y==2)//2
		{
			while(true)//整体同上 
			{
				if(q2.empty()){
					cout<<-1<<" ";
					break;
				}
				if(!vis[q2.top().m])
				{
					cout<<q2.top().n<<" ";
					vis[q2.top().m]=true;
					q2.pop();
					break;
				}
				else q2.pop();
			}
		}
		if(y==3)//3
		{
			while(true)//整体同上 
			{
				if(q3.empty()){
					cout<<-1<<" ";
					break;
				}
				if(!vis[q3.top().m])
				{
					cout<<q3.top().n<<" ";
					vis[q3.top().m]=true;
					q3.pop();
					break;
				}
				else q3.pop();
			}
		}
	}
	return 0;//完成 
}

ok!结束战斗！

大佬轻喷