前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 1101. 彼此熟识的最早时间(排序+并查集)

LeetCode 1101. 彼此熟识的最早时间(排序+并查集)

作者头像
Michael阿明
发布2020-07-13 14:47:14
8550
发布2020-07-13 14:47:14
举报

1. 题目

在一个社交圈子当中,有 N 个人。每个人都有一个从 0 到 N-1 唯一的 id 编号。

我们有一份日志列表 logs,其中每条记录都包含一个非负整数的时间戳,以及分属两个人的不同 id,logs[i] = [timestamp, id_A, id_B]

每条日志标识出两个人成为好友的时间,友谊是相互的:如果 A 和 B 是好友,那么 B 和 A 也是好友。

如果 A 是 B 的好友,或者 A 是 B 的好友的好友,那么就可以认为 A 也与 B 熟识。

返回圈子里所有人之间都熟识的最早时间。如果找不到最早时间,就返回 -1 。

代码语言:javascript
复制
示例:
输入:logs = [[20190101,0,1],[20190104,3,4],[20190107,2,3],[20190211,1,5],
		[20190224,2,4],[20190301,0,3],[20190312,1,2],[20190322,4,5]],
	 N = 6
输出:20190301
解释:
第一次结交发生在 timestamp = 20190101,0 和 1 成为好友,
社交朋友圈如下 [0,1], [2], [3], [4], [5]。
第二次结交发生在 timestamp = 20190104,3 和 4 成为好友,
社交朋友圈如下 [0,1], [2], [3,4], [5].
第三次结交发生在 timestamp = 20190107,2 和 3 成为好友,
社交朋友圈如下 [0,1], [2,3,4], [5].
第四次结交发生在 timestamp = 20190211,1 和 5 成为好友,
社交朋友圈如下 [0,1,5], [2,3,4].
第五次结交发生在 timestamp = 20190224,2 和 4 已经是好友了。
第六次结交发生在 timestamp = 20190301,0 和 3 成为好友,大家都互相熟识了。
 

提示:

1 <= N <= 100
1 <= logs.length <= 10^4
0 <= logs[i][0] <= 10^9
0 <= logs[i][1], logs[i][2] <= N - 1
保证 logs[i][0] 中的所有时间戳都不同
Logs 不一定按某一标准排序
logs[i][1] != logs[i][2]

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/the-earliest-moment-when-everyone-become-friends 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

参考数据结构:并查集

  • 先按时间排序
  • 按时间顺序合并两个人,检查是否只有一个团,如果是,返回当前时间
代码语言:javascript
复制
class dsu
{
	vector<int> f;
public:
	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);
		int fb = find(b);
		f[fa] = fb;
	}
	int find(int a)
	{
		int origin = a;
		while(a != f[a])
			a = f[a];
		return f[origin] = a;
	}
	bool onlyOne()
	{
		int count = 0;
		for(int i = 0; i < f.size(); ++i)
		{
			if(i == find(i))
				count++;
			if(count > 1)
				return false;
		}
		return true;
	}
};
class Solution {
public:
    int earliestAcq(vector<vector<int>>& logs, int N) {
    	sort(logs.begin(), logs.end(),[&](auto a, auto b){
    		return a[0] < b[0];
    	});
    	dsu u(N);
    	for(auto& lg : logs)
    	{
    		u.merge(lg[1], lg[2]);
    		if(u.onlyOne())
    			return lg[0];
    	}
    	return -1;
    }
};

244 ms 32.4 MB

  • 或者,有效的合并 N-1次就完成了
代码语言:javascript
复制
class dsu
{
	vector<int> f;
public:
	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);
		int fb = find(b);
		f[fa] = fb;
	}
    bool weAreFriend(int a, int b)
    {
        int fa = find(a);
		int fb = find(b);
        return fa == fb;
    }
	int find(int a)
	{
		int origin = a;
		while(a != f[a])
			a = f[a];
		return f[origin] = a;
	}
};
class Solution {
public:
    int earliestAcq(vector<vector<int>>& logs, int N) {
    	sort(logs.begin(), logs.end(),[&](auto a, auto b){
    		return a[0] < b[0];
    	});
    	dsu u(N);
    	for(auto& lg : logs)
    	{
            if(!u.weAreFriend(lg[1], lg[2]))
            {	//不是朋友时,才进行合并
                u.merge(lg[1], lg[2]);
                N--;
                if(N==1)//合并了N-1次,找到了
                    return lg[0];
            }
        }
    	return -1;
    }
};

240 ms 32.3 MB

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

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

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

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

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