DS实验题 Old_Driver UnionFindSet结构

题目

思路

很典型的并查集问题，朋友A和B要合并到一个统一的集合中，也就是Union(A, B)操作，在Union操作中需要先找到A所属的集合的代表元和B所属集合的代表元，也就是使用Find(A)和Find(B)，如果它们的代表元不一样，则将B集合合并到A集合(pre[B_root] = A_root)。

在Find的过程中，使用了路径压缩算法，使用i记录当前节点，使用j记录当前i的上级：

1.用j记录i的上级 2.将i的上级改为i_root 3.i=j

本题中，另外一种关系，也就是敌人，可以用一个二维数组存储，或者用邻接表存储都可以。

代码：

//
//  main.cpp
//  UnionFindSet
//
//  Created by wasdns on 16/12/15.
//  Copyright © 2016年 wasdns. All rights reserved.
//

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

int pre[1000];

int rela[105][105];

void Initial(int n)
{
    memset(pre, 0, sizeof(pre));
    memset(rela, 0, sizeof(rela));

    for (int i = 1; i <= n; i++)
    {
        pre[i] = i;
    }
}

int Find(int a)
{
    int fa = a;

    while (fa != pre[fa])
    {
        fa = pre[fa];
    }

    int i = a, j;

    while (i != fa)     //路径压缩算法。
    {
        j = pre[i];     //记录上级
        pre[i] = fa;    //修改当前上级为root
        i = j;          //来到原来的上级
    }

    return fa;
}

void Union(int a, int b)
{
    int afa = Find(a);
    int bfa = Find(b);

    if (afa != bfa)
    {
        pre[bfa] = afa; //b所属集合的名字，改为a所属集合的名字
    }
}

void CreatRela(int m)
{
    int i;

    int x, y, r;

    for (i = 1; i <= m; i++)
    {
        cin >> x >> y >> r;

        if (r == 1) Union(x, y);

        else if (r == -1) {

            rela[x][y] = -1;

            rela[y][x] = -1;
        }
    }
}

void Query(int k)
{
    int i;

    int x, y;

    for (i = 1; i <= k; i++)
    {
        cin >> x >> y;

        if (rela[x][y] == 0)
        {
            if (Find(x) == Find(y)) {
                cout << "Good job" << endl;
            }

            else {
                cout << "No problem" << endl;
            }
        }

        else
        {
            if (Find(x) == Find(y)) {
                cout << "OK but..." << endl;
            }

            else {
                cout << "No way" << endl;
            }
        }
    }
}

int main()
{
    int n, m, k;

    cin >> n >> m >> k;

    Initial(n);

    CreatRela(m);

    Query(k);

    return 0;
}

/*
 7 8 4
 5 6 1
 2 7 -1
 1 3 1
 3 4 1
 6 7 -1
 1 2 1
 1 4 1
 2 3 -1
 */