Leetcode原题链接:链表相交
一、一般题解
尽量低的时间、空间复杂度,一开始想到的就是先算长度,然后长链表节点先追上短链表节点,使剩余长度相等,然后再同步后移找到相交节点,Python代码如下:
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> ListNode:
if headA == None or headB is None:
return None
curA, curB = headA, headB
lenA, lenB = 0, 0
while curA != None:
curA = curA.next
lenA += 1
while curB != None:
curB = curB.next
lenB += 1
gap = abs(lenA - lenB)
curA, curB = headA, headB
if lenA > lenB:
while gap != 0:
curA = curA.next
gap -= 1
elif lenA < lenB:
while gap != 0:
curB = curB.next
gap -= 1
while curA != curB:
curA = curA.next
curB = curB.next
return curA
时间复杂度O(n),空间复杂度O(1)。
*一些需要注意的点:
- 考虑两链表均为空的情况
- 由于无论两链表谁长谁短或是等长,均需要以下代码,故可以挪到最后减少代码量。
while curA != curB:
curA = curA.next
curB = curB.next
二、神仙解法
当成追击问题,两个节点分别从两链表头节点出发,遍历完当前链表后,去遍历另一条链表,直到二者相等,Python代码如下:
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> ListNode:
if headA == None or headB is None:
return None
curA, curB = headA, headB
while curA != curB:
if curA != None:
curA = curA.next
else:
curA = headB
if curB != None:
curB = curB.next
else:
curB = headA
return curA
*一些没有立刻理解的点:
- 相同长度但无相同节点时,岂不是永远追不上?答:从代码中可以看出,A、B两链表均会遍历至空节点即
None
,故这种情况下,相同节点即为空节点,返回None
,符合题意。 - 为什么不能遍历完自己的链表继续遍历自己的,直到追上对方?答案:不知道,尝试这么提交,但是失败了,失败原因超出时间限制,失败示例中两链表是1~20000的奇数偶数。搞不懂为什么用”二“的方法就能解决。