D. pigeon swap
题目链接:https://atcoder.jp/contests/abc395/tasks/abc395_d
题意:
三个操作:1.移动鸽子a到巢穴b
2.交换巢穴a和巢穴b的鸽子
3.查询鸽子a在哪个巢穴中
思路:
开三个数组,分别记录鸽子i的位置,巢穴i的位置,位置i的巢穴
#include<bits/stdc++.h>
#define rep(i,a,n) for(int i=a;i<=n;i++)
#define pb push_back
#define endl "\n"
#define fi first
#define se second
//#pragma GCC optimize(3)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
const int inf=0x3f3f3f3f;
const ll llmax=LLONG_MAX;
const int maxn=1e6+5;
const int mod=1e9+7;
int n,q;
int e[maxn];//记录第i个鸽子在哪个巢
int f[maxn];//记录第i个巢穴在哪个位置
int g[maxn];//记录第i个位置是哪个巢穴
signed main()
{
ios::sync_with_stdio(false),cin.tie(0);
cin>>n>>q;
rep(i,1,n){
e[i]=i;
f[i]=i;
g[i]=i;
}
while(q--){
int opt;cin>>opt;
if(opt==1){
int a,b;cin>>a>>b;
e[a]=f[b];
}else if(opt==2){
int a,b;cin>>a>>b;
swap(f[a],f[b]);
g[f[a]]=a;
g[f[b]]=b;
}else{
int a;cin>>a;
cout<<g[e[a]]<<endl;
}
}
return 0;
}

浙公网安备 33010602011771号