首页
学习
活动
专区
圈层
工具
发布
社区首页 >专栏 >LeetCode 2092. 找出知晓秘密的所有专家(并查集)

LeetCode 2092. 找出知晓秘密的所有专家(并查集)

作者头像
Michael阿明
发布2022-01-07 10:52:56
发布2022-01-07 10:52:56
4970
举报

文章目录

代码语言:txt
复制
- [1. 题目](https://cloud.tencent.com/developer)
- [2. 解题](https://cloud.tencent.com/developer)

1. 题目

给你一个整数 n ,表示有 n 个专家从 0 到 n - 1 编号。

另外给你一个下标从 0 开始的二维整数数组 meetings ,其中 meetings[i] = [xi, yi, timei] 表示专家 xi 和专家 yi 在时间 timei 要开一场会。

一个专家可以同时参加 多场会议 。最后,给你一个整数 firstPerson 。

专家 0 有一个 秘密 ,最初,他在时间 0 将这个秘密分享给了专家 firstPerson

接着,这个秘密会在每次有知晓这个秘密的专家参加会议时进行传播。更正式的表达是,每次会议,如果专家 xi 在时间 timei 时知晓这个秘密,那么他将会与专家 yi 分享这个秘密,反之亦然。

秘密共享是 瞬时发生 的。也就是说,在同一时间,一个专家不光可以接收到秘密,还能在其他会议上与其他专家分享。

在所有会议都结束之后,返回所有知晓这个秘密的专家列表。你可以按 任何顺序 返回答案。

代码语言:javascript
复制
示例 1:
输入:n = 6, meetings = [[1,2,5],[2,3,8],[1,5,10]], firstPerson = 1
输出:[0,1,2,3,5]
解释:
时间 0 ,专家 0 将秘密与专家 1 共享。
时间 5 ,专家 1 将秘密与专家 2 共享。
时间 8 ,专家 2 将秘密与专家 3 共享。
时间 10 ,专家 1 将秘密与专家 5 共享。
因此,在所有会议结束后,专家 0、1、2、3 和 5 都将知晓这个秘密。

示例 2:
输入:n = 4, meetings = [[3,1,3],[1,2,2],[0,3,3]], firstPerson = 3
输出:[0,1,3]
解释:
时间 0 ,专家 0 将秘密与专家 3 共享。
时间 2 ,专家 1 与专家 2 都不知晓这个秘密。
时间 3 ,专家 3 将秘密与专家 0 和专家 1 共享。
因此,在所有会议结束后,专家 0、1 和 3 都将知晓这个秘密。

示例 3:
输入:n = 5, meetings = [[3,4,2],[1,2,1],[2,3,1]], firstPerson = 1
输出:[0,1,2,3,4]
解释:
时间 0 ,专家 0 将秘密与专家 1 共享。
时间 1 ,专家 1 将秘密与专家 2 共享,专家 2 将秘密与专家 3 共享。
注意,专家 2 可以在收到秘密的同一时间分享此秘密。
时间 2 ,专家 3 将秘密与专家 4 共享。
因此,在所有会议结束后,专家 0、1、2、3 和 4 都将知晓这个秘密。

示例 4:
输入:n = 6, meetings = [[0,2,1],[1,3,1],[4,5,1]], firstPerson = 1
输出:[0,1,2,3]
解释:
时间 0 ,专家 0 将秘密与专家 1 共享。
时间 1 ,专家 0 将秘密与专家 2 共享,专家 1 将秘密与专家 3 共享。
因此,在所有会议结束后,专家 0、1、2 和 3 都将知晓这个秘密。
 
提示:
2 <= n <= 10^5
1 <= meetings.length <= 10^5
meetings[i].length == 3
0 <= xi, yi <= n - 1
xi != yi
1 <= timei <= 10^5
1 <= firstPerson <= n - 1

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/find-all-people-with-secret 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

参考:数据结构 并查集

  • 按时间对会议进行排序
  • 对相同时间进行的会议进行并查集合并操作,之后再次检查,对不知道秘密的专家,撤销合并操作(之前不知道这个,导致一直超时)
代码语言:javascript
复制
class dsu{
public:
    vector<int> f;
    dsu(int n)
    {
        f.resize(n);
        for(int i = 0; i < n; ++i)
            f[i] = i;
    }
    void merge(int a, int b)
    {
        int fa = find(a), fb = find(b);
        f[fa] = fb;
    }
    int find(int a)
    {
        if(a == f[a]) return a;
        return f[a] = find(f[a]);
    }
    void unmerge(int a)
    {
        f[a] = a;
    }
};
class Solution {
public:
    vector<int> findAllPeople(int n, vector<vector<int>>& meetings, int firstPerson) {
        int mlen = meetings.size();
        vector<bool> known(n, false);
        known[0] = known[firstPerson] = true;
        sort(meetings.begin(), meetings.end(),[&](auto& a, auto& b){
            return a[2] < b[2];
        });
        dsu uset(n);
        for(int i = 0; i < mlen; )
        {
            int t = meetings[i][2], j = i;
            while(j < mlen && meetings[j][2]==t)
            {   // 时间相同的会议
                int a = meetings[j][0], b = meetings[j][1];
                int f1 = uset.find(a);
                int f2 = uset.find(b);
                if(known[f2]) // b 知道秘密
                    uset.merge(a, b);//有顺序的,注意!!!
                else
                    uset.merge(b, a);
                j++;
            }
            for(int k = i; k < j; ++k)
            {
                int a = meetings[k][0], b = meetings[k][1];
                int f1 = uset.find(a);
                int f2 = uset.find(b);
                if(known[f1] || known[f2])
                    known[a]=known[b]=true;
                else
                {	// 两人都不知道秘密,恢复他们的初始状态
                    uset.unmerge(a);
                    uset.unmerge(b);
                }
            }
            i = j;
        }
        vector<int> ans;
        for(int i = 0; i < n; ++i)
            if(known[i])
                ans.push_back(i);
        return ans;
    }
};

600 ms 130.9 MB C++


我的CSDN博客地址 https://michael.blog.csdn.net/

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2021/11/28 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 文章目录
  • 1. 题目
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档