题目来源
https://leetcode.cn/problems/intersection-of-two-linked-lists-lcci/
题目描述
给你两个单链表的头节点 headA 和 headB ,请你找出并返回两个单链表相交的起始节点。如果两个链表没有交点,返回 null 。
思路
我们求出两个链表的长度,并求出两个链表长度的差值,然后让curA移动到,和curB 末尾对齐的位置,
此时我们就可以比较curA和curB是否相同,如果不相同,同时向后移动curA和curB,如果遇到curA == curB(这里不是当前节点的值相等,而是以cur为头结点的链表相等),则找到交点。
‘’’
代码
class ListNode:
def __init__(self, x):
self.val = x
self.next = None
class Solution:
def get_size(self, head: ListNode): #计算链表长度
cur = head
i = 0
while cur != None:
i += 1
cur=cur.next
return i
def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> ListNode:
sizeA=self.get_size(headA)
sizeB=self.get_size(headB)
n=abs(sizeA-sizeB)
if sizeB>=sizeA:
curA=headB
curB=headA
else:
curA=headA
curB=headB
for _ in range(n):# 让curA和curB在同一起点上(末尾位置对齐)
curA=curA.next
while curA!=None: # 让curA和curB在同一起点上(末尾位置对齐)
if curA==curB:
return curA
else:
curA=curA.next
curB=curB.next
return None