【Leetcode】2092. Find All People With Secret

题目地址：

https://leetcode.com/problems/find-all-people-with-secret/

有$n$个人，编号为$0,1,...,n-1$。某个人$r,1\le r\le n-1$被$0$号人告诉了一个秘密。接着，开了$m$次会议，每次会议以三元组$(x,y,t)$表示，意思是$t$时刻$x$和$y$开了个会，并且每次开会的时候，知道秘密的人会把秘密分享给所有和他参加过同一场会议的人（如果某个时刻，某个知道秘密的人同时参与了多场会议，则所有参加了这些会议的人，以及所有参加了包含这些人的会议的人，都知道了秘密。例如，如果$(1,2,3),(2,3,3),(3,4,3)$，假设$3$时刻之前，$1$知道了秘密，那么$3$时刻之后，$1,2,3,4$就都知道了秘密）。问开完所有会议之后，知道秘密的有哪些人。

思路是并查集。先将$r$和$0$合并起来。接着，对会议按时间排序，然后遍历这些会议，每次遍历的时候，关注某一时刻开的所有会议，将每个会议的两个人union起来。此时，知道秘密的人都合并到了同一个集合里，除此之外的所有人都是不知道秘密的。接着，将这个时刻的所有两个人都不知道秘密的会议里，两个人的祖宗节点都设为自己（因为过了这个时刻，这些没能获知秘密的人不会再散播秘密了，下个时刻做合并的时候，他们将作为一个单独的集合去合并）。遍历完之后，统计一下哪些点和$0$同属一个集合即可。代码如下：

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class Solution {
    int[] p;
    
    public List<Integer> findAllPeople(int n, int[][] meetings, int first) {
        p = new int[n];
        for (int i = 0; i < n; i++) {
            p[i] = i;
        }
        
        Arrays.sort(meetings, (x, y) -> Integer.compare(x[2], y[2]));
        // first和0是知道秘密的
        p[first] = 0;
        for (int i = 0; i < meetings.length; i++) {
            int j = i;
            while (j < meetings.length && meetings[j][2] == meetings[i][2]) {
            	// 把参会的两个人合并到一起
                int x = meetings[j][0], y = meetings[j][1];
                p[find(x)] = find(y);
                j++;
            }
            // 此时，知道秘密的人已经合并到一起，不知道秘密的人也合并到了一起
            int p0 = find(0);
            j = i;
            while (j < meetings.length && meetings[j][2] == meetings[i][2]) {
            	// 如果某次会议是由两个不知道秘密的人参加的，那么他们将作为独立的个体参加后面的会议
                int x = meetings[j][0], y = meetings[j][1];
                if (find(x) != p0 && find(y) != p0) {
                    p[x] = x;
                    p[y] = y;
                }
                j++;
            }
            i = j - 1;  
        }
        
        List<Integer> res = new ArrayList<>();
        int p0 = find(0);
        for (int i = 0; i < n; i++) {
            if (find(i) == p0) {
                res.add(i);
            }
        }
        
        return res;
    }
    
    int find(int x) {
        if (x != p[x]) {
            p[x] = find(p[x]);
        }
        return p[x];
    }
}

时间复杂度$O(m\log m+(m+n){\log }^{\ast }n)$，空间$O(n)$。