剑指 Offer 52. 两个链表的第一个公共节点（双指针/set()）

Posted 2020-12-19 Yelush

• 题目描述

输入两个链表，找出它们的第一个公共节点。

如下面的两个链表：



在节点 c1 开始相交。

 

示例 1：



输入：intersectVal = 8, listA = [4,1,8,4,5], listB = [5,0,1,8,4,5], skipA = 2, skipB = 3
输出：Reference of the node with value = 8
输入解释：相交节点的值为 8 （注意，如果两个列表相交则不能为 0）。从各自的表头开始算起，链表 A 为 [4,1,8,4,5]，链表 B 为 [5,0,1,8,4,5]。在 A 中，相交节点前有 2 个节点；在 B 中，相交节点前有 3 个节点。

• 解法一：p1和p2指针依次往后遍历

class Solution:
    ‘‘‘
    剑指offer题解：假设长的链表是p1，短的是p1，先让长的走长度差然后和p1和p1依次往后遍历，直到p1=p2为止
    ‘‘‘
    def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> ListNode:
        lengthHa, lengthHb = 0, 0
        p1 = headA
        p2 = headB
        while headA:
            lengthHa += 1
            headA = headA.next
        while headB:
            lengthHb += 1
            headB = headB.next
        if lengthHb >= lengthHa:
            p1, p2 = p2, p1
        nlength = abs(lengthHa - lengthHb)
        for i in range(nlength):
            p1 = p1.next
        while p1 and p2 and p1 != p2:
            p1 = p1.next
            p2 = p2.next
        return p1

• 解法二：遍历两次链表/链表交换遍历

class Solution:
    ‘‘‘
    大神题解：遍历两次链表，假如headA长度为5，headB长度为8，用两个指针同时遍历headA和headB,当短的遍历完了则指针交，这样第二次遍历的长度交换为8和5，这样则同步了。
    ‘‘‘
    def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> ListNode:
        p1 = headA
        p2 = headB
        while p1 != p2:
            p1 = p1.next if p1 else headB
            p2 = p2.next if p2 else headA
        return p1

• 解法三：巧用Python的set结构(没有重复值)，不易想到

class Solution:
    ‘‘‘
    大神题解：用set，set存的数组是不带重复的，将headA和headB都存到set里面去，当set长度不再递增时，说明此时重复了
    ‘‘‘
    def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> ListNode:
        saveSet = set()
        while headA:
            saveSet.add(headA)
            headA = headA.next
        lengthSet = len(saveSet)
        while headB:
            saveSet.add(headB)
            if len(saveSet) ==lengthSet:
                return headB
            else:
                lengthSet += 1
                headB = headB.next
        return None