Codeforces 1382 E. Mastermind —— 想法，贪心

This way

题意：

现在给你一个长度为n的数组a，让你构造一个新的数组b使得b和a位置相同且数相同的位置有x个，数的multiset集合的交集大小为y。

题解：

那么首先由于数量多的数对于y-x的限制比较大，那么我们就优先将数量最多的数当成x。注意这里是每次数量最多，而不是一开始数量最多。优先队列Q就是做这个的。
队列p[i]表示值为i的所有数的位置
接下来找y-x的数是哪些。还是从数量多的开始找，注意剩下的位置有可能是小于这个数的数量的，那么就要取个min。最后放y-x个数的话，就循环移位放就行了，用并查集的话会出错。
ned[i]表示第i个数在y-x中有多少

#include<bits/stdc++.h>
using namespace std;
#define pa pair<int,int>
const int N=1e5+5;
queue<int>p[N];
struct node{
    int v,num;
    bool operator< (const node& a)const {
        return num>a.num;
    }
}a[N];
int ned[N],ans[N],vis[N],b[N];
priority_queue<pa>Q;
int main()
{
    int t;
    scanf("%d",&t);
    while(t--){
        int n,x,y,v;
        scanf("%d%d%d",&n,&x,&y);
        for(int i=1;i<=n+1;i++){
            while(!p[i].empty())p[i].pop();
            a[i]={i,0},ned[i]=0,ans[i]=0,vis[i]=0;
            fa[i]=i;
        }
        while(!Q.empty())Q.pop();
        for(int i=1;i<=n;i++)
            scanf("%d",&b[i]),a[b[i]].num++,Q.push({a[b[i]].num,b[i]}),p[b[i]].push(i),vis[b[i]]=1;
        int res=y-x;
        if(res<0){
            printf("NO\n");
            continue;
        }
        for(int i=1;i<=x;i++){
            v=Q.top().second;Q.pop();
            int pos=p[v].front();p[v].pop();
            ans[pos]=v;
            a[v].num--;
        }
        sort(a+1,a+1+n+1);
        int pos=1;
        while(pos<=n&&a[pos].num&&res){
            int can=min(res,min(n-x-a[pos].num,a[pos].num));
            ned[a[pos].v]=can;
            res-=can;
            pos++;
        }
        if(res){
            printf("NO\n");
            continue;
        }
        int f=0;
        pos=1;
        for(int i=1;i<=n&&a[i].num;i++){
            if(!ned[a[i].v])continue;
            while(ned[a[i].v]){
                if(pos==i||!p[a[pos].v].size())pos++;
                if(!a[pos].num)pos=1;
                if(pos==i){
                    f=1;
                    break;
                }
                while(ned[a[i].v]&&p[a[pos].v].size())
                    ans[p[a[pos].v].front()]=a[i].v,ned[a[i].v]--,p[a[pos].v].pop();
            }
            if(ned[a[i].v]||f){
                f=1;
                break;
            }
        }
        if(f){
            printf("NO\n");
            continue;
        }
        int no=1;
        while(vis[no])no++;
        printf("YES\n");
        for(int i=1;i<=n;i++)
            printf("%d%c",ans[i]?ans[i]:no," \n"[i==n]);
    }
    return 0;
}
/*
2
6 3 6
1 1 1 2 2 3
*/