题目介绍

团伙问题（并查集） 在城市里住着n个人，任何两个认识的人不是朋友就是敌人，而且满足： (1)朋友的朋友是朋友； (2)敌人的敌人是朋友。 这n个人可以划分为若干个团伙，使得每个团伙中任意两个成员均为朋友。告诉你关于这n个人的m条信息，即某两个人是朋友，或者某两个人是敌人，请你编写一个程序，计算这个城市最多可能有多少个团伙？ Input 输入包含多组数据，对于每组数据： 第1行为n和m，1<n<1000,1<=m<=100000； 以下m行，每行为p、x、y，p的值为0或1，p为0时，表示x和y是朋友，p为1时，表示x和y是敌人。 Output： 对于每组输入数据，输出一个整数，表示这n个人最多可能有几个团伙。 Sample Input 6 4 1 1 4 0 3 5 0 4 6 1 1 2 Sample Output 4

题目思路

利用并查集来维护团伙关系，初始是每个人都是一个团伙，然后遍历输入的每一条关系并处理，如果是朋友则两个团伙合并并做路径压缩，同时团伙数减一。

具体代码

package com.dreamchaser;


import java.util.Scanner;

/**
 * 团伙问题（并查集）
 */
public class test6 {

    /**
     * 记录团伙关系
     */
    static int[] persons;

    static int number;
    /**
     * 人数
     */
    static int n;
    /**
     * 关系数
     */
    static int m;

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);

        n = scanner.nextInt();

        m = scanner.nextInt();
        persons=new int[n+1];
        number=n;


        int p,x,y;
        for (int i = 0; i < m; i++) {
            p=scanner.nextInt();
            x=scanner.nextInt();
            y=scanner.nextInt();
            if (p==0){
                if (merge(x,y)!=0){
                    number--;
                }
            }
        }
        System.out.println(number);
    }






    /**
     * 寻找根节点
     *
     * @param i
     * @return
     */
    private static int findUnion(int i) {
        /**
         * 合并集的根
         */

        if (persons[i] != 0) {
            int root = findUnion(persons[i]);
            //路径压缩
            persons[i] = root;
            return root;
        } else {
            return i;
        }
    }

    /**
     * 合并集合
     * 如果两个属于同一个集合则返回null，否则返回根节点
     * @param x
     * @param y
     * @return
     */
    private static int merge(int x, int y) {
        int root1 = findUnion(x);
        int root2 = findUnion(y);
        if (root1 != root2) {
            persons[root2] = root1;
            return root1;
        } else {
            return 0;
        }

    }
}

运行截图

记录点滴，乐于分享，转载请注明出处。

愿我们能以梦为马，不负人生韶华！ 与君共勉！