Golang判断两个链表是否相交的方法详解

// 定义链表节点
type Node struct {
	val  int
	next *Node
}

// 判断两个链表是否相交
func IsIntersect(list1, list2 *Node) bool {
	if list1 == nil || list2 == nil {
		return false
	}

	m := make(map[*Node]struct{})
	p := list1
	for p != nil {
		m[p] = struct{}{}
		p = p.next
	}

	p = list2
	for p != nil {
		if _, ok := m[p]; ok {
			return true
		}
		p = p.next
	}

	return false
}

// 根据数组生成链表
func New(data []int) *Node {
	nodes := make([]*Node, len(data))
	for i := 0; i < len(data); i++ {
		nodes[i] = &Node{
			val: data[i],
		}
		if i > 0 {
			nodes[i].next = nodes[i-1]
		}
	}

	return nodes[len(data)-1]
}

// 合并两个链表
func Connect(node1, node2 *Node) *Node {
	if node1 == nil {
		return node2
	}

	if node2 == nil {
		return node1
	}

	p := node1
	for p.next != nil {
		p = p.next
	}

	p.next = node2
	return node1
}

func main() {
	data1 := []int{1, 2, 3, 4, 5}
	data2 := []int{6, 7, 8, 9, 10}
	data3 := []int{11, 12, 13, 14, 15}

	node1 := New(data1)
	node2 := New(data2)
	node3 := New(data3)

	node2 = Connect(node2, node1)  // 10,9,8,7,6,5,4,3,2,1
	node3 = Connect(node3, node1)  // 15,14,13,12,11,5,4,3,2,1

	result := data0312.IsIntersect(node2, node3)
	fmt.Println(result) // true
}

func IsIntersect(list1, list2 *Node) bool {
	if list1 == nil || list2 == nil {
		return false
	}

	// 将链表1的尾指向头
	p := list1
	for p.next != nil {
		p = p.next
	}
	p.next = list1

	// 遍历链表2，如果能到达链表1的头则说明相交
	p = list2
	for p != nil {
		if p == list1 {
			return true
		}

		p = p.next
	}

	return false
}