【Leetcode题解——链表】面试题 02.07. 链表相交

面试题 02.07. 链表相交

先计算长度是因为两个指针同时比较会增加复杂度,而我们需要的是直接从较短的链表开始比较(若链表相交则之后的节点都相等),在两个链表长度相等时可以同时比较

【Leetcode题解——链表】面试题 02.07. 链表相交_第1张图片 

算法流程:

  1. 计算两个链表的长度
  2. 计算长度的差值dif,较长的链表先移动差值dif步,此时两个链表的长度相等
  3. 两个链表一起移动,当指针相等时(不是值相等)返回该指针,若无相等的情况则返回none
# 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) -> ListNode:
        lena,lenb=0,0
        cur_a,cur_b=headA,headB
        #计算a,b两个链表的长度
        while cur_a:
            cur_a=cur_a.next
            lena+=1
        while cur_b:
            cur_b=cur_b.next
            lenb+=1
        cur_a,cur_b=headA,headB
        #较长的链表先移动abs(len(b)-len(a))步
        if lena>lenb:
            dif=lena-lenb
            for _ in range(dif):
                cur_a=cur_a.next
        elif lena

你可能感兴趣的:(leetcode,链表,算法,数据结构,python,leetcode)