思路:用vector储存每个点可以交换的点,然后深搜看能否通过交换得到现有的序列,即判断这两个点是否可以连通。
代码如下:
#include<iostream>
#include<vector>
#include<cstring>
using namespace std;
int target[110];//储存目标序列
vector<int>v[110];
bool vis[110][110];//已访问标记数组
bool dfs(int st,int cur,int ed){//深搜
vis[st][cur]=1;
if(cur==ed)return 1;//如果通过深搜找到了目标点,即连通
for(int i=0;i<v[cur].size();i++){
if(!vis[st][v[cur][i]]){
if(dfs(st,v[cur][i],ed))return 1;
}
}
return 0;
}
int main(){
int n,i,j,d;
cin>>n;
for(i=1;i<=n;i++)
cin>>target[i];
for(i=1;i<=n;i++){
v[i].push_back(i);
cin>>d;
if(i+d<=n){
v[i].push_back(i+d);//i可以连通到i+d
v[i+d].push_back(i);//因为是无向图,i+d也可以联通到i
}
if(i-d>=1){
v[i].push_back(i-d);//i可以连通到i-d
v[i-d].push_back(i);//因为是无向图,i-d也可以联通到i
}
}
int flag1=1;
for(i=1;i<=n;i++){
if(!dfs(i,i,target[i]))flag1=0;//如果有一个点与无法交换到目标的点,即该点与目标点不连通,就输出NO
}
if(flag1)cout<<"YES"<<endl;
else cout<<"NO"<<endl;
return 0;
}