如何判断两个链表是否相交

45 阅读1分钟

算法题:判断2个链表相交

面试中可能会问到的算法题,今天总结一下

方法一:map

步骤:

  • 1.遍历list1,以节点为key放入map中
  • 2.遍历list2,判断每个节点是否在map中,如果在则相交,且顶一个存在的节点是交点
// 定义链表节点
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
}

方法二:首尾相接法

将链表1的尾指向头,然后遍历链表2,看是否能达到链表1的头,如果能则说明相交

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
}

参考

[1]如何判断两个单链表(无环)是否交叉