D - 香蕉戏

D - 香蕉戏

题意：给定一个含有n个元素的数列，第i号元素开始时数值为i,元素i可以与距离为d[i]的元素进行交换。问经过任意次交换能不能得到给定数列。
这题我们直接拿样例来看：
由：1 2 3 4 5 6 7
到：4 3 5 1 2 7 6
d[i]: 4 6 6 1 6 6 1
先看1–>4的转换过程
先是4号元素和5号元素换
1 2 3 5 4 6 7
再是1号元素和5号元素换
4 2 3 5 1 6 7
再是4号元素和5号元素换
4 2 3 1 5 6 7
发现中间过程借助了5号元素
是不是觉得很神奇，也很熟悉！！！
借助5号元素实现了1号元素和4号元素的连通（交换）…
这不就是求连通性问题吗？！！
那么，不就可以用并查集和Floyd来写了
肯定都看懂了，不废话了，上菜
AC code：
并查集：

#include <bits/stdc++.h>

using namespace std;

const int N=105;
int a[N],f[N],d[N];

int Find(int k)
{
    if(f[k]==k) return k;
    return f[k]=Find(f[k]);
}

int merg(int x,int y)
{
    x=Find(x),y=Find(y);
    if(x!=y) f[y]=x;
}

int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        f[i]=i;
    }
    for(int i=1;i<=n;i++)
    {
        cin>>d[i];
        if(i+d[i]<=n) merg(i,i+d[i]);
        if(i>d[i]) merg(i-d[i],i);
    }
    for(int i=1;i<=n;i++)
    {
        if(Find(i)!=Find(a[i]))
        {
            cout<<"No";
            return 0;
        }
    }
    cout<<"YES";

    return 0;
}

Floyd：

#include <bits/stdc++.h>

using namespace std;

const int N=105;
int a[N],f[N][N];

int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=n;i++)
    {
        f[i][i]=1;
        int d;
        cin>>d;
        if(i+d<=n) f[i][i+d]=f[i+d][i]=1;
        if(i>d) f[i][i-d]=f[i-d][i]=1;
    }
    for(int k=1;k<=n;k++)
    {
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                if(f[i][k] && f[k][j]) f[i][j]=f[j][i]=1;
            }
        }
    }
    for(int i=1;i<=n;i++)
    {
        if(!f[i][a[i]])
        {
            cout<<"NO";
            return 0;
        }
    }
    cout<<"YES";

    return 0;
}