题目:输出俩个链表的相交节点,如果没有则返回空值
解法:与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