Codeforces 1283C Friends and Gifts

题目链接：

Codeforces 1283C Friends and Gifts

思路：

将还没有分配的人和还没有被分配的人分别存进两个数组，挨个遍历，如果相同位置是同一个人，就和下一个交换

代码：

#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> P;
typedef long long ll;
#define fi first
#define sc second
#define pb(a) push_back(a)
#define mp(a,b) make_pair(a,b)
#define pt(a) cerr<<a<<"---\n"
#define rp(i,n) for(int i=0;i<n;i++)
#define rpn(i,n) for(int i=1;i<=n;i++)
const int maxn=2e5+5;
int f[maxn];
bool flag[maxn];
int main(){
	ios::sync_with_stdio(false); cin.tie(nullptr);
	int n; cin>>n;
	rpn(i,n){
		cin>>f[i]; flag[f[i]]=true;
	}
	vector<int> a,b;
	rpn(i,n){
		if(f[i]==0) a.pb(i);
		if(flag[i]==false) b.pb(i);
	}
	int sz=a.size();
	for(int i=0;i<sz;i++){
		if(a[i]==b[i]){
			int x=i,y=(i+1)%sz;
			swap(b[x],b[y]);
		}
	}
	rp(i,sz) f[a[i]]=b[i];
	rpn(i,n) cout<<f[i]<<' ';
	return 0;
}