【图论】测试题A - pSort

题意

给定一个从1到n的数组，给一个目标数组和d[i]，第i个数只能和距离d[i]的元素交换，要求判断是否能变换成目标数组

思路

直接用并查集的思路去解决

代码

#include<map>
#include<cmath>
#include<stack>
#include<queue>
#include<cstdio>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#define Endl "\n"
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int maxn = 1e3+5;
const int INF = 0x3f3f3f;
const int mod = 1e9 + 7;
const double pi = acos(-1); 
inline ll read(){
	ll x = 0, f = 1; char ch; 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;
}
int n;
bool flag;
int a[maxn],b[maxn],t[maxn];

void makeset()
{
	for(int i=1;i<=n;i++)
		t[i]=i;
}

int getf(int x)
{
	if(x!=t[x])
		t[x]=getf(t[x]);
	return t[x];
}

void unionset(int x,int y)
{
	int k=getf(x);
	int v=getf(y);
	if(k!=v)
	{
		t[v]=k;
	}
	return ;
}
void solve() {
	flag=1;
	n=read();
	makeset();
	for(int i=1;i<=n;i++)
		a[i]=read();
	for(int i=1;i<=n;i++)
	{
		b[i]=read();
		if(i>b[i])
			unionset(i,i-b[i]);
		if(i+b[i]<=n)
			unionset(i,i+b[i]);
//		cout<<"----tsst------"<<Endl;
//		for(int i=1;i<=n;i++)
//			cout<<t[i]<<" ";
//		cout<<Endl;
	}
	for(int i=1;i<=n;i++)
	{
		if(getf(t[i])!=getf(a[i]))
		{
			flag=0;
		}
	}
	
	if(flag)	cout<<"YES"<<Endl;
	else	cout<<"NO"<<Endl;
	return ; 
} 

int main() {
//	for(int T = read(); T; T--)
		solve();
	return 0;
}