腾讯笔试题——朋友圈（并查集、Golang）

YTIANYE

已于 2022-03-05 22:23:56 修改

阅读量1k

点赞数

分类专栏：算法文章标签：并查集社交网络最大群体算法 Golang

于 2022-03-05 22:02:13 首次发布

本文链接：https://blog.csdn.net/qq_41577750/article/details/123302143

版权

算法专栏收录该内容

16 篇文章 0 订阅

订阅专栏

例题


链接：朋友圈(后端开发卷)_腾讯笔试题_牛客网
来源：牛客网
 

现在有 105 个用户，编号为 1- 105，现在已知有 m 对关系，每一对关系给你两个数 x 和 y ，代表编号为 x 的用户和编号为 y 的用户是在一个圈子中，例如： A 和 B 在一个圈子中， B 和 C 在一个圈子中，那么 A , B , C 就在一个圈子中。现在想知道最多的一个圈子内有多少个用户。

数据范围： 1≤m≤2×106 1 \le m \le 2 \times 10^6 \ 1≤m≤2×106 
进阶：空间复杂度 O(n) O(n) \ O(n)  ，时间复杂度 O(nlogn) O(nlogn) \ O(nlogn) 

输入描述:

第一行输入一个整数T，接下来有T组测试数据。
对于每一组测试数据：第一行输入1个整数n，代表有n对关系。
接下来n行，每一行输入两个数x和y，代表编号为x和编号为y的用户在同一个圈子里。

1 ≤ T ≤ 10

1 ≤ n ≤ 2*106

1 ≤ x, y ≤ 105


 

输出描述:

对于每组数据，输出一个答案代表一个圈子内的最多人数
示例1

输入
2
4
1 2
3 4
5 6
1 6
4
1 2
3 4
5 6
7 8
输出
4
2

解析

并查集：参考【算法】并查集（Disjoint Set）[共3讲]_哔哩哔哩_bilibili好久没有发算法的视频了，今天补上一段讲“并查集”的视频。并查集是一种用来合并不相交集合的数据结构（虽然我在视频里面写得非常不像数据结构）。一个比较常用的地方是：用来检测一个图里面是不是包含有环。这个结构在Kruskal算法（最小生成树）里面有非常重要的作用。因为很久没有用C语言了，这次就直接用C语言来实现。大家如果有用其他语言实现的版本的话，欢迎发上来分享。https://www.bilibili.com/video/BV13t411v7Fs?p=1

代码（Golang）


package main

import "fmt"

// 并查集

type Set struct {
	roots  map[int]int
	height map[int]int
}

func (this Set) append(x, y int) {
	xRoot := this.findRoot(x)
	yRoot := this.findRoot(y)

	if xRoot == 0 && yRoot == 0 {
		this.roots[x] = y
		this.roots[y] = y // 根指向自己
		this.height[y] = 1
	} else if xRoot == 0 && yRoot != 0 {// 注意判断条件写全
		this.roots[x] = yRoot
	} else if yRoot == 0 && xRoot != 0{
		this.roots[y] = xRoot
	} else {
		if this.height[xRoot] >= this.height[yRoot] {
			this.roots[yRoot] = xRoot                                          // 注意是yRoot的根

			// for k, v := range this.roots{
			// 	if v == yRoot{
			// 		this.roots[k] = xRoot
			// 	}
			// }

			this.height[xRoot] = max(this.height[xRoot], this.height[yRoot]+1) // 注意是xRoot的高度根据xRoot和yRoot的高度发生改变
		} else {
			this.roots[xRoot] = yRoot // 注意一个根指向另一个根

			// for k, v := range this.roots{
			// 	if v == xRoot{
			// 		this.roots[k] = yRoot
			// 	}
			// }
		}
	}
}

func max(a, b int) int {
	if a > b {
		return a
	}
	return b
}

// 找父节点
func (this Set) findRoot(x int) int {
	for x != this.roots[x] {
		x = this.roots[x]
	}
	return x
}

// 最多结点的集合
func (this Set) maxCount() int {
	count := make(map[int]int)
	for k := range this.roots {
		count[this.findRoot(k)]++ // 注意这里的findRoot
	}

	// for _, v := range this.roots{
	// 	count[v]++
	// }

	res := 0
	for _, v := range count {
		res = max(res, v)
	}
	return res
}

func main() {

	var T int
	fmt.Scan(&T)
	for ; T != 0; T-- {
		set := Set{map[int]int{}, map[int]int{}}
		var n int
		fmt.Scan(&n)
		for ; n != 0; n-- {
			var x, y int
			fmt.Scan(&x, &y)
			set.append(x, y)
		}
		fmt.Println(set.maxCount())
	}
}

样例