CodeForces - 1283C Friends and Gifts

原题链接：

https://vjudge.net/problem/CodeForces-1283C

AC代码：

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <list>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <stdio.h>
#include <stdlib.h>
using namespace std;
const int INF = 0x3fffffff;
typedef long long ll;
typedef unsigned long long ull;

void __init__()
{
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
} //head

const int maxn = 202108;

int f[maxn];
int rec[maxn];
int main()
{
    // __init__();
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        cin >> f[i];
        rec[f[i]] = 1;
    }
    set<int> a;
    set<int> b;
    set<int> c;
    set<int> d;
    for (int i = 1; i <= n; i++)
    {
        if (!f[i])
        {
            a.insert(i);
        }
        if (!rec[i])
        {
            b.insert(i);
        }
    }
    for (auto t : a)
    {
        if (b.count(t))
        {
            //相同的
            c.insert(t);
        }
        else
        {
            //不同的
            d.insert(t);
        }
    }
    for (auto t : c)
    {
        b.erase(t);
    }
    if (c.size() == 1)
    {
        int ttt;
        ttt = *c.begin();
        f[ttt] = *b.begin();
        b.erase(b.begin());
        b.insert(ttt);
    }

    set<int>::iterator sd = d.begin();
    set<int>::iterator sb = b.begin();
    for (int i = 0; i < d.size(); i++)
    {
        f[*sd] = *sb;
        sd++;
        sb++;
    }
    int temp[maxn];
    int cc = 0;
    for (auto y : c)
    {
        temp[cc++] = y;
    }
    if (c.size() != 1)
    {
        for (int i = 0; i < cc; i++)
        {
            if (i == cc - 1)
            {
                f[temp[i]] = temp[0];
            }
            else
            {
                f[temp[i]] = temp[i + 1];
            }
        }
    }

    for (int i = 1; i <= n; i++)
    {
        if (i == 1)
        {
            cout << f[i];
        }
        else
        {
            cout << " " << f[i];
        }
    }
    cout << endl;

    return 0;
}

用了很麻烦的方法做的，好歹也A了，大体就是找出空位和待填入的数，将相同的部分取出，剩余的不同的部分对着填就行了，相同的部分循环赋值，首位与末尾相对应，这么做得考虑一种情况，就是当相同的部分只有一个数的时候，要做一些特殊处理，见代码中if (c.size() == 1)部分。