38/300
- 相交链表
编写一个程序,找到两个单链表相交的起始节点。
如下面的两个链表:
在节点 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 个节点。
示例 2:
输入:intersectVal = 2, listA = [0,9,1,2,4], listB = [3,2,4], skipA = 3, skipB = 1
输出:Reference of the node with value = 2
输入解释:相交节点的值为 2 (注意,如果两个列表相交则不能为 0)。从各自的表头开始算起,链表 A 为 [0,9,1,2,4],链表 B 为 [3,2,4]。在 A 中,相交节点前有 3 个节点;在 B 中,相交节点前有 1 个节点。
示例 3:
输入:intersectVal = 0, listA = [2,6,4], listB = [1,5], skipA = 3, skipB = 2
输出:null
输入解释:从各自的表头开始算起,链表 A 为 [2,6,4],链表 B 为 [1,5]。由于这两个链表不相交,所以 intersectVal 必须为 0,而 skipA 和 skipB 可以是任意值。
解释:这两个链表不相交,因此返回 null。
注意:
如果两个链表没有交点,返回 null.
在返回结果后,两个链表仍须保持原有的结构。
可假定整个链表结构中没有循环。
程序尽量满足 O(n) 时间复杂度,且仅用 O(1) 内存。
这道题看完例1,百思不得其解
为啥不是1开始呢?!看解释里的输入输出,有点明白了。
如果输入
8 [4,1,8,4,5] [5,0,1,8,4,5] 2 3
那就是从8开始相交。
如果输入:
1 [4,1,8,4,5] [5,0,1,8,4,5] 1 2
那就是从1开始相交,相交节点前有几个节点是会被考虑进去的。
但是,默认头只给了两个argument,估计是它后台自己考虑吧。
class Solution(object):
def getIntersectionNode(self, headA, headB):
"""
:type head1, head1: ListNode
:rtype: ListNode
"""
朴素思路:遍历二位算长度,然后咔嚓去掉长出来的那些节点,从一边齐开始比。
取巧思路:短的到头了就补上长的,长的到头了就补上短的,互补后这一轮肯定一边儿齐,如图:
class Solution(object):
def getIntersectionNode_01(self, headA, headB):
"""
:type head1, head1: ListNode
:rtype: ListNode
"""
if not headA or not headB: return None
l, r = headA, headB
countA, countB = 0, 0
while headA:
countA +=1
headA = headA.next
while headB:
countB +=1
headB = headB.next
differ = abs(countA - countB)
for i in range(differ):
if countA > countB: l = l.next
else: r = r.next
while l != r:
l, r = l.next, r.next
return l
def getIntersectionNode_02(self, headA, headB):
"""
:type head1, head1: ListNode
:rtype: ListNode
"""
if not headA or not headB: return None
l, r = headA, headB
while l != r:
l = headB if not l else l.next
r = headA if not r else r.next
#找到相同的退出循环(None也算相同的)
return l