【短学期算法作业】团伙问题(并查集)

题目介绍

团伙问题(并查集)
在城市里住着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;
        }

    }
}

运行截图

在这里插入图片描述


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

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

猜你喜欢

转载自blog.csdn.net/qq_46101869/article/details/108508544