模拟/贪心 Codeforces Round #699 (Div. 2) C题

模拟/贪心 Codeforces Round #699 (Div. 2) C题

题意：给定初始颜色序列和最终目标颜色序列，再给定m个人，每个人必须按顺序给某一个格子染上ci的颜色，问是否可以染成目标序列。

解析：
将操作序列倒着考虑，
如果有要染成c[i]这个颜色的，直接将它染成c[i]，前面的人随意修改不影响
如果没有要染成c[i]这个颜色的，就看是否是第一次染色
如果不是第一次染色，则让res[i]=last就好，不影响
如果是第一次染色，则看最终目标序列里有这个颜色，
如果有这个颜色，就让res[i]=pos[c[i]]
不然，就代表不能染色，无解，break
最后，如果num不为0，则代表没有染成目标颜色序列，无解

stack<int> cnt[N];
int a[N],b[N],c[N],pos[N],res[N],vis[N];
int main(){
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        int num = 0;
        for(int i=1;i<=n;i++) {
            scanf("%d",&b[i]);
            if(a[i] != b[i]){
                if(cnt[b[i]].empty()) num++;
                cnt[b[i]].push(i);
            }
            pos[b[i]] = i;//存颜色最后出现的下标
        }
        for(int i=1;i<=m;i++) scanf("%d",&c[i]);
        int last = 0;
        int flag = 1;
        for(int i=m;i>=1;i--){
            //如果没有要改成c[i]这个颜色的
            if(cnt[c[i]].empty()){
                if(!last){
                    if(pos[c[i]]) res[i] = pos[c[i]];
                    else{
                        flag = 0;
                        break;
                    }
                }else{
                    res[i] = last;
                }
            }else{
                res[i] = cnt[c[i]].top();
                cnt[c[i]].pop();
                if(cnt[c[i]].empty()) num--;
            }
            last = res[i];
        }
        if(flag && !num){
            puts("YES");
            for(int i=1;i<=m;i++) printf("%d ", res[i]);
            puts("");
        }else{
            puts("NO");
        }
        for(int i=1;i<=n;i++) {
            vis[i]=0;
            while(cnt[i].size()) cnt[i].pop();
        }
        for(int i=1;i<=n;i++) pos[i]=0;
    }
    return 0;
}