1481C Fence Painting (list、map)

题目

思路:
对于失败有两种情况:
1.统计所有的需要修正成各种颜色的数目,如果发现对于一种颜色我们所拥有的画师数少于所需修正的数目,则失败。
2.我们会发现其实可能有画师会画不属于b[i]中所有的颜色,如果这个画师是最后一位画师(最后画)那么没有任何的画师可以在之后将他的画修改为正确颜色,失败。

再来分析成功的画师分配:对于每个需要修改颜色的地方用队列做位置记录,分配画师的时候队列不空就pop掉一个,空了的话就(也就是这种画师该修改的颜色位置已经全部修正完毕,该画师多余)直接修改最后一个画师的画,反正最后一个画师会将其修正为正确颜色,还要小心分析一下最后一个画师的位置,如果最后一个画师所画颜色所需数量和画师数量一样,那这个最后一个画师所画位置=队列中此颜色最后一个push的,如果最后一个画师是剩余的那么找到一个匹配其颜色的位置即可。

#include<iostream>
#include<memory.h>
#include<queue>
#include<map>
using namespace std;
typedef long long ll;
const int Max = 1e5 + 5;
int h[Max], b[Max];
int lst[Max], ls[Max];
queue<int> q[Max];

int main()
{
    
    
	int t;cin >> t;
	while (t--)
	{
    
    
		int n, m;cin >> n >> m;
		map<int, int> mag, mah, ma;
		for (int i = 1;i <= n;i++) {
    
     cin >> lst[i];ma[lst[i]]++; }
		for (int i = 1;i <= n;i++) {
    
     cin >> ls[i];ma[ls[i]]++; }
		for (int i = 1;i <= m;i++) {
    
     cin >> h[i];mah[h[i]]++;ma[h[i]]++; }
		for (auto i = ma.begin();i != ma.end();i++)q[i->first] = queue<int>();
		for (int i = 1;i <= n;i++)
		{
    
    
			if (lst[i] != ls[i])
			{
    
    
				q[ls[i]].push(i);
				mag[ls[i]]++;
			}
		}
		int f = 0, last = 0;
		for (auto i = mag.begin();i != mag.end();i++)
		{
    
    
			if (mah[i->first] < i->second) {
    
     f = 1;break; }
		}
		for (int i = 1;i <= n;i++)if (ls[i] == h[m])last = i;
		if (mah[h[m]] == mag[h[m]])last = q[h[m]].back();
		if (last == 0)f = 1;
		if (f)cout << "NO" << endl;
		else
		{
    
    
			cout << "YES" << endl;
			for (int i = 1;i <= m;i++)
			{
    
    
				if (!q[h[i]].empty())
				{
    
    
					cout << q[h[i]].front()<<" ";q[h[i]].pop();
				}
				else cout << last<<" ";
			}
			cout << endl;
		}
	}
}

猜你喜欢

转载自blog.csdn.net/asbbv/article/details/113748106