Hlg 1748 【并查集】.cpp

题意：

　　n个数编号从1~n..但是并不按顺序排列..

　　现在想把他们互换位置最后得到1~n的序列..

　　两个数可以交换位置的条件是 |i-j| == bi

　　

　　输入：n 表示n个数

　　　　a[1], a[2], a[3]..a[i]..a[n]

　　　　b[1], b[2], b[3]..b[i]..b[n]

思路：

　　并查集..

　　可以这么看..如果两个数在一个集合里，那么这两个数就可以互相交换位置了..

　　所以就按照i+-b[i]把可以交换的位置的数和该数的归在一个集合里..

　　最后查看是否所有要改的数都在一个集合里..

Tips：

　　加的时候加的应该是a[i], 表示把这些数放在一个集合里..

　　最后看是不是要改位置的数正好都在一个集合里..

　　如果把下标放在一个集合里就没有意义了..

Code：

　　

#include <stdio.h>
#include <cstring>

int f[110];
int find(int x)
{
    return f[x] == x?x:f[x] = find(f[x]);
}

int n;
int a[110], b[110];

int main()
{
    freopen("in.txt", "r", stdin);
    bool flag;
    while (~scanf("%d", &n)) {
        flag = true;
        for (int i = 1; i <= n; ++i)
            f[i] = i;
        for (int i = 1; i <= n; ++i)
            scanf("%d", &a[i]);
        for (int i = 1; i <= n; ++i)
            scanf("%d", &b[i]);
        for (int i = 1; i <= n; ++i) {
            int ff = find(a[i]), fa, fb;
            if (i+b[i] < n) fa = find(a[i+b[i]]);
            if(i-b[i] > 0) fb = find(a[i-b[i]]);
            if (i+b[i] < n && fa != ff) {
                f[fa] = ff;
            }
            if (i-b[i] > 0 && fb != ff) {
                f[fb] = ff;
            }
        }
        for (int i = 1; i <= n; ++i) {
            if (find(i) != find(a[i])) {
               flag = false;
               break;
            }
        }
        if (flag) puts("YES");
        else puts("NO");

    }
    return 0;
}

链接：http://acm.hrbust.edu.cn/index.php?m=ProblemSet&a=showProblem&problem_id=1748