相交链表2,python解法

题目:输出俩个链表的相交节点,如果没有则返回空值

解法:与1不同的是,该题没有说明不存在环形链表,所以第一步要先判断是否是环形链表,是的话返回第一个入环节点,然后进行讨论。

有三种情况:

1、如果链表1和链表2都不是环形链表,那就跟上篇文章的解法一样了,双指针法;

2、如果链表1为环形链表,链表2不是环形链表,那么它们将不可能相交,返回空;

3、如果链表1和链表2都为环形链表,那么会有以下三种情况:

        1、俩个环形链表不相交;

 

        2、俩个环形链表先相交,再入环。

        3、俩个环形链表先入环,再相交。

接下来用代码来讨论这三种情况。

# 输出俩个链表的第一个相交节点,如果没有返回空值
# class ListNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution(object):
    def getIntersectionNode(self, headA, headB):
        """
        :type head1, head1: ListNode
        :rtype: ListNode
        """
        if not headA or not headB: return None
        # 先判断是不是环形链表,是则返回入环的第一个节点,不是返回空
        # 获取链表1的入环节点
        loop1 = self.detectCycle(headA)
        # 获取链表2的入环节点
        loop2 = self.detectCycle(headB)
        # 情况1,如果链表1和链表2都不是环形链表
        if not loop1 and not loop2:
            return self.noloop(headA, headB)
        # 情况2,都为环形链表
        if loop1 and loop2:
            return self.bothloop(headA, loop1, headB, loop2)
        # 情况3,一个为环形链表,一个不是,俩个链表不可能相交,返回空
        return
    def detectCycle(self, head):
        """
        :type head: ListNode
        :rtype: ListNode
        """
        # 快慢指针
        slow, fast = head, head
        while True:
            # 如果为空则说明没有环,返回空
            if not fast or not fast.next: return
            # 快指针走俩步,慢指针走一步,最后相遇则证明有环
            fast = fast.next.next
            slow = slow.next
            # 相遇退出循环
            if fast == slow: break
        # 快指针返回到头节点
        fast = head
        while fast != slow:
            # 快满指针同时走一步,最后再次相遇则为入环节点
            fast = fast.next
            slow = slow.next
        return fast

    def noloop(self, head1, head2):
        # 链表1和链表2都不是环形链表,只有俩种可能
        # 1.俩个链表不相交 2.俩个链表相交
        A = head1
        B = head2
        # 双指针,当A和B相等时即可返回相交节点,没有相交节点则返回空
        while A != B:
            A = A.next if A else head2
            B = B.next if B else head1
        return A

    def bothloop(self, head1, loop1, head2, loop2):
        # 当入环节点相同时,证明这俩个链表是先相交变一个链表再一起入环的
        # 用noloop方法即可,把loop值替代做空值
        if loop1 == loop2:
            A = head1
            B = head2
            while A != B:
                A = A.next if A == loop1 else head2
                B = B.next if B == loop2 else head1
            return A
        # 当入环节点不相同时,有俩种情况,1.俩个环形链表不相交,2.链表先入环再相交,此时返回任意一个入环节点都可以
        else:
            cur1 = loop1.next
            # 链表1先入环跑一圈
            while cur1 != loop1:
                # 当这一圈还没跑完就遇到loop2,证明这俩个链表是相交的
                if cur1 == loop2:
                    return loop1
                cur1 = cur1.next
            # 跑完一圈没有遇到,说明俩个链表不相交
            return

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值