给你两个单链表的头节点 headA 和 headB ,请你找出并返回两个单链表相交的起始节点。如果两个链表不存在相交节点,返回 null 。
图示两个链表在节点 c1 开始相交:
题目数据保证整个链式结构中不存在环。
注意,函数返回结果后,链表必须保持其原始结构 。
自定义评测:
评测系统 的输入如下(你设计的程序不适用此输入):
intersectVal - 相交的起始节点的值。如果不存在相交节点,这一值为 0
listA - 第一个链表
listB - 第二个链表
skipA - 在 listA 中(从头节点开始)跳到交叉节点的节点数
skipB - 在 listB 中(从头节点开始)跳到交叉节点的节点数
评测系统将根据这些输入创建链式数据结构,并将两个头节点 headA 和 headB 传递给你的程序。如果程序能够正确返回相交节点,那么你的解决方案将被视作正确答案 。
题目链接:https://leetcode.cn/problems/intersection-of-two-linked-lists
分析
如果两个链表相交,短的链表会先走到相交点,相交点之后的长度是相同的;相交点之前的长度可能不同,两条链表走到末尾,各自指向对方的头节点可以消除长度差,最终在相交点相遇
实现
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> Optional[ListNode]:
pa = headA
pb = headB
while pa != pb:
pa = pa.next if pa else headB
pb = pb.next if pb else headA
return pa
指针 pa 指向 A 链表,指针 pb 指向 B 链表,依次往后遍历
如果 pa 到了末尾,则 pa = headB 继续遍历
如果 pb 到了末尾,则 pb = headA 继续遍历
如果存在相交点,pA和pB会在相交的处相遇(pa == pb)
如果不存在相交点,pA和pB会一直遍历完另一条链表,最终指向null