golang 实现并查集

原创于 2021-07-21 19:29:48 发布 · 308 阅读

0 ·

CC 4.0 BY-SA版权

go 专栏收录该内容

1 篇文章

订阅专栏

该博客介绍了一种使用并查集算法来检测图中环路的方法。通过初始化数据，遍历图的边并应用`union`函数判断是否形成环路，如果检测到环路则输出提示并退出程序，否则表明图中没有环路。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package main

import (
	"fmt"
	"os"
)

const SIZE =6


//找到根节点
func findRoot(x int,data []int)int  {
	root :=x
	if data[root]==-1{
		return root
	} else {
		root = findRoot(data[root],data)
		return root
	}
}

// 0 表示有环了
func union(x int,y int,data []int,rank []int) int {
	xRoot := findRoot(x,data)
	yRoot := findRoot(y,data)
	if xRoot ==yRoot{
		return 0
	}else {
		if rank[xRoot]>rank[yRoot]{
			data[yRoot] = xRoot
			rank[xRoot] +=1
		}else {
			data[xRoot] = yRoot
			rank[yRoot] +=1
		}
		return 1
	}

}

func initData(data []int,rank []int){
	for index:=0;index<SIZE;index++{
		data[index] =-1
		rank[index] =0
	}
}

func main() {
	data := make([]int,SIZE)
	rank :=make([]int,SIZE)
	initData(data,rank)
	graph :=[6][2]int{
		{0,1},{1,2},{1,3},{3,4},{2,5},
		{5,4},
	}
	for i:=0;i<SIZE;i++{
		res := union(graph[i][0],graph[i][1],data,rank)
		if res==0{
			fmt.Println("cycle detect")
			os.Exit(1)
		}
	}
	fmt.Println("no cycle")


}