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

面试题 02.07. 链表相交

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

 

算法流程：

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<lenb:
            dif=lenb-lena
            for _ in range(dif):
                cur_b=cur_b.next
        #再一起移动，直到指针相等
        while cur_a:
            if cur_a==cur_b:
                return cur_a
            cur_a=cur_a.next
            cur_b=cur_b.next
        return None