Codeforces Round #733 (Div. 1 + Div. 2, based on VK Cup 2021 - Elimination (Engine)) D. Secret Santa

每年12月,VK都会为员工举办一场名为“秘密圣诞老人”的活动。事情是这样发生的。

编号为1 ~𝑛的员工𝑛参加活动。每个员工𝑖都被分配给一个不同的员工𝑏𝑖,员工𝑖必须给该员工做一份新年礼物。每个员工恰好被分配给另一个员工,没有人被分配给自己(但两个员工可能被分配给彼此)。形式上,所有𝑏𝑖必须是1和𝑛之间不同的整数,对于任何𝑖,𝑏𝑖≠𝑖必须成立。

作业通常是随机生成的。今年,作为一个实验,所有的活动参与者都被问及他们希望给谁做礼物。每个员工𝑖都说他们想给员工𝑎𝑖做一份礼物。

找到一个有效的任务𝑏,最大限度地满足员工的愿望。

输入
每个测试包含多个测试用例。第一行包含测试用例的数量𝑡(1≤𝑡≤105)。测试用例的描述如下。

每个测试用例由两行组成。第一行包含单个整数𝑛(2≤𝑛≤2⋅105)——该事件的参与者人数。

第二行包含𝑛整数𝑎1𝑎2,…,𝑎𝑛(1≤𝑎𝑖≤𝑛;𝑎𝑖≠𝑖)——员工的愿望从1到𝑛。

保证𝑛在所有测试用例上的总和不超过2⋅105。

输出
对于每个测试用例,打印两行。

在第一行中,打印单个整数𝑘(0≤𝑘≤𝑛)—任务中实现的愿望的数量。

在第二行,打印𝑛截然不同的整数𝑏1𝑏2,…,𝑏𝑛(1≤𝑏𝑖≤𝑛;𝑏𝑖≠𝑖)-分配给员工1、2、…、𝑛的员工数量。

𝑘必须等于𝑖的值的数量,这样𝑎𝑖=𝑏𝑖,并且必须尽可能大。如果有多个答案,打印任何一个。

例子
inputCopy
2
3.
2 1 2
7
6 4 6 2 4 5 6
outputCopy
2
3 1 2
4
6 4 7 2 3 5 1
请注意
在第一个测试用例中,存在两个有效的赋值:[3,1,2]和[2,3,1]。前一个作业满足两个愿望,而后一个作业只满足一个愿望。因此,𝑘=2,唯一正确的答案是[3,1,2]。

思路:不变动最多,变动最少,因为有相同的,我们可以直接记录第一次出现在符合位置的,这些都是可以不变动的,变动的位置我们可以打上标记,后续改动可以分辨出,哪些可以移动。出现符合的位置用map标记一下,然后我们可以遍历,如果没有标记的,就是没有出现的,要往里面放的,我们这里用vector存起来。之后便有两种情况,一种是只有一个没有放进去的,这种情况又有两种可能,一种是可以直接放进去,我们可以直接放进去输出;另一种是需要替换,因为是一个没有放进去,那么肯定是有一个数字是重复的,两个位置都可以放这个重复的数字,所以我们把两个交换下位置放进去即可;第二种情况是有许多个没有放进去的,我们可以用队列,来反复放,如果能放就放,这个到最后也有可能出现,最后一个数字刚好等于他的下标放不进去,我们可以仿照第一种情况的方法,任意找一个符合的非固定的交换,再放入即可。

代码:

#include <iostream>
#include <algorithm>
#include <string.h>
#include <string>
#include <math.h>
#include <stdio.h>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<tuple>
#include<numeric>
#include<stack>
using namespace::std;
typedef long long  ll;
int n,t;
inline __int128 read(){
    __int128 x = 0, f = 1;
    char ch = getchar();
    while(ch < '0' || ch > '9'){
        if(ch == '-')
            f = -1;
        ch = getchar();
    }
    while(ch >= '0' && ch <= '9'){
        x = x * 10 + ch - '0';
        ch = getchar();
    }
    return x * f;
}
inline void print(__int128 x){
    if(x < 0){
        putchar('-');
        x = -x;
    }
    if(x > 9)
        print(x / 10);
    putchar(x % 10 + '0');
}

int w;
int a[200005];
int b[200005];
void solv(){
    map<int,int>q;
    vector<int>we;
    cin>>n;
    int na=0;
    for (int i =0; i<=n; i++) {
        a[i]=0;
        b[i]=0;
    }
    for (int i =1; i<=n; i++) {
        cin>>w;
        if (w!=i&&!q.count(w)) {
            b[i]=1;
            a[i]=w;
            na++;
            q[w]=1;
            continue;
        }
        if (q[w]) {
            q[w]++;
        }
    }
//    for(auto [x,y]:q){
//        printf("%d %d\n",x,y);
//    }
    int kl=0;
    queue<int>ff;
    for (int i =1; i<=n; i++) {
        if (!q[i]) {
            ff.push(i);
            kl++;
        }
    }
//    while (!ff.empty()) {
//        printf("%d ",ff.front());ff.pop();
//    }printf("\n");
//    return;
    printf("%d\n",na);
    if (kl==0) {
        for (int i =1; i<=n; i++) {
            printf("%d ",a[i]);
        }printf("\n");return;
    }
    if (kl==1) {
        int bj=0;
        for (int i =1; i<=n; i++) {
            if (a[i]==0&&ff.front()!=i) {
                a[i]=ff.front();
                bj=1;
            }
        }
//        printf("%d\n",bj);
        if (bj) {
            for (int i =1; i<=n; i++) {
                printf("%d ",a[i]);
            }printf("\n");return;
        }
        else{
            int rme;
            int bj2=0;
            for (int i=1; i<=n; i++) {
                if (q[i]==2&&ff.front()!=i&&!bj2) {
                    rme=i;
                    bj2=1;
                }
            }
//            printf("reme::::%d\n",rme);
            for (int i =1; i<=n; i++) {
                if (a[i]==rme) {
                    printf("%d ",ff.front());
                    continue;
                }
                if (a[i]==0) {
                    printf("%d ",rme);continue;
                }
                printf("%d ",a[i]);
            }printf("\n");
        }
        return;
    }
    else{
//        printf("n:::%d\n",n);
        int kjkkj=0;
        for (int i =1; i<=n; i++) {
            
//            printf("%d\n",kjkkj);
//            printf("SDasd\n");
            if (a[i]==0) {
                while (ff.front()==i) {
                    int ls=ff.front();
                    ff.pop();ff.push(ls);
                }
                a[i]=ff.front();
                ff.pop();
                if (ff.size()==1) {
                    break;
                }
                
            }
        }
        
        int bj=0;
        for (int i =1; i<=n; i++) {
            if (a[i]==0&&ff.front()!=i) {
                a[i]=ff.front();
                bj=1;
            }
        }
//        printf("%d\n",bj);
        if (bj) {
            for (int i =1; i<=n; i++) {
                printf("%d ",a[i]);
            }printf("\n");return;
        }
        else{
            int rme;
//            int bj2=0;
//            for (int i=1; i<=n; i++) {
//                if (&&ff.front()!=i&&!bj2) {
//                    rme=i;
//                    bj2=1;
//                }
//            }
//            printf("reme::::%d\n",rme);
            int bjee=1;
            for (int i =1; i<=n; i++) {
                if (b[i]==0&&bjee) {
                    bjee=0;
                    rme=a[i];
                    printf("%d ",ff.front());
                    continue;
                }
                if (a[i]==0) {
                    printf("%d ",rme);continue;
                }
                printf("%d ",a[i]);
            }printf("\n");
        }
        
    }
   
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(); cout.tie();
    cin>>t;
    while (t--) {
        solv();
    }
    return 0;
}
 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值