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;
}


posted @ 2025-03-02 09:51  Marinaco  阅读(33)  评论(0)    收藏  举报
//雪花飘落效果