CF1481C Fence Painting

C. Fence Painting

本题细节较多，因此我在 45min 通过时，只有 600 人左右通过。

首先容易想到倒序考虑，因为后来者的颜色肯定在上面。我们可以记录一下 \(a,b\) 中有多少颜色需要更换，有哪些颜色不需要更换。我们在考虑颜色时，首先考虑的就是那些需要更改的颜色，如果现在要考虑的颜色已经不再需要了，就选择下一个要涂的人所涂的那个栅栏即可，这样下一个人会盖过现在的颜色。

细节贼多，注意数组清空。

//Don't act like a loser.
//This code is written by huayucaiji
//You can only use the code for studying or finding mistakes
//Or,you'll be punished by Sakyamuni!!!
#include<bits/stdc++.h>
#define int long long
using namespace std;

int read() {
	char ch=getchar();
	int f=1,x=0;
	while(ch<'0'||ch>'9') {
		if(ch=='-')
			f=-1;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9') {
		x=x*10+ch-'0';
		ch=getchar();
	}
	return f*x;
}

const int MAXN=1e5+10;

int n,m,a[MAXN],b[MAXN],c[MAXN],ans[MAXN],tot;
stack<int> cnt[MAXN];
int ex[MAXN];

signed main() {
	int t=read();
	while(t--) {
		n=read();m=read();
		tot=0;
		for(int i=1;i<=n;i++) {//细节，清空数组
			int sz=cnt[i].size();
			while(sz--) {
				cnt[i].pop();
			}
			ex[i]=0;
		}
		for(int i=1;i<=n;i++) {
			a[i]=read();
		}
		for(int i=1;i<=n;i++) {
			b[i]=read();
			if(b[i]!=a[i]) {
				if(cnt[b[i]].empty()) {
					tot++;//有多少种颜色需要更改
				}
				cnt[b[i]].push(i);
			}
			ex[b[i]]=i;//记录那些颜色出现
		}
		for(int i=1;i<=m;i++) {
			c[i]=read();
		}
		
		int last=0;
		bool flag=1;
		for(int i=m;i;i--) {
			if(!cnt[c[i]].empty()) {
				ans[i]=cnt[c[i]].top();
				cnt[c[i]].pop();
				if(cnt[c[i]].empty()) {
					tot--;
				}
			}
			else {
				if(last==0) {
                    //即考虑最后一个粉刷匠，单独处理
					if(ex[c[i]]) {
						ans[i]=ex[c[i]];
					}
					else {
                        //没地方画，输出NO
						printf("NO\n");
						flag=0;
						break;
					}
				}
				else {
					ans[i]=last;
				}
			}
			last=ans[i];
		}
		if(flag&&tot) {
            //若颜色不够用，没有改好，输出NO
			printf("NO\n");
		}
		else if(flag){
			printf("yEs\n");
			for(int i=1;i<=m;i++) {
				printf("%lld ",ans[i]);
			}
			cout<<endl;
		}
	}
	return 0;
}