LeetCode:160(Python)—— 相交链表(简单)

相交链表

概述:给你两个单链表的头节点 headA headB ,请你找出并返回两个单链表相交的起始节点。如果两个链表不存在相交节点,返回 null

输入:intersectVal = 8, listA = [4,1,8,4,5], listB = [5,6,1,8,4,5], skipA = 2, skipB = 3
输出:Intersected at '8'

输入:intersectVal = 2, listA = [1,9,1,2,4], listB = [3,2,4], skipA = 3, skipB = 1
输出:Intersected at '2'

输入:intersectVal = 0, listA = [2,6,4], listB = [1,5], skipA = 3, skipB = 2
输出:null

方法一:双指针

思路:只有当链表 headAheadB 都不为空时,两个链表才可能相交。因此首先判断链表 headA headB 是否为空,如果其中至少有一个链表为空,则两个链表一定不相交,返回 null。当链表 headA headB 都不为空时,创建两个指针 A B,初始时分别指向两个链表的头节点 headA headB,然后将两个指针依次遍历两个链表的每个节点。

# 双指针
class Solution:
    def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> Optional[ListNode]:
        A = headA
        B = headB
        while A and B:
            if A == B:
                return A
            A = A.next
            B = B.next
            if A is None and B is None:
                return None
            if A is None:
                A = headB
            if A is None:
                B = headA
# 另外一种写法
class Solution:
    def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> Optional[ListNode]:
        A, B = headA, headB
        while A != B:
            A = A.next if A else headB
            B = B.next if B else headA
        return A

总结

错的人就算走过了对方的路也还是会错过,这题我希望大家都返回 True !

你可能感兴趣的:(1024程序员节,python,职场和发展,leetcode,链表)