题目
给你两个单链表的头节点 headA 和 headB ,请你找出并返回两个单链表相交的起始节点。如果两个链表没有交点,返回 null 。
图示两个链表在节点 c1 开始相交:
题目数据 保证 整个链式结构中不存在环。
注意,函数返回结果后,链表必须 保持其原始结构 。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/intersection-of-two-linked-lists-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
1.对于两个单链表,其中有一个为空则不存在交集
2.不空的链表,如果有交集,最后一个元素必为交集中的元素,所以可以通过遍历两个链表看各个链表末尾的元素是否为同一个,如果是则存在交集,这时候需要考虑第一个相交的位置;如果不是一个元素,则说明不存在交集
3.问题来到了如何求第一个交集元素的位置,如果对于两个长度相同的链表,依次从头遍历,只需要找到首个指针指向相同位置处的元素即可。而对于长度不等的情况,例如len(a)>len(b),就需要让长的那个链表a先走len(a)-len(b)步,这样两者就等价于等长链表了,依次寻找当前位置对应元素是否相同,不同都往后走一步,直到两者指向同一个位置,这个位置处的元素就是所求结果。
代码
class Solution:
def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> ListNode:
pa,pb=headA,headB
sa,sb=headA,headB
len_a,len_b=0,0
if not pa or not pb:
return None
while pa.next:
len_a+=1
pa=pa.next
while pb.next:
len_b+=1
pb=pb.next
if pa!=pb:
return None
if len_a>len_b:
while len_a>len_b:
sa=sa.next
len_a-=1
elif len_a<len_b:
while len_a<len_b:
sb=sb.next
len_b-=1
while sa!=sb:
sa=sa.next
sb=sb.next
return sa