链表——3.链表相交

力扣题目链接

给你两个单链表的头节点 headA 和 headB ，请你找出并返回两个单链表相交的起始节点。如果两个链表没有交点，返回 null 。

示例：

输入：listA = [4,1,8,4,5]，listB = [5,0,1,8,4,5]
输出：8

题干的意思很简单，即给你两个链表，判断这两个链表在某一个节点是否相交，如图所示。

两个链表在交点在8。具体思路如《代码随想录》所示。

完整代码如下：

class Solution:
    def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> ListNode:
        lenA, lenB = 0, 0
        cur = headA
        while cur:         # 求链表A的长度
            cur = cur.next 
            lenA += 1
        cur = headB 
        while cur:         # 求链表B的长度
            cur = cur.next 
            lenB += 1
        curA, curB = headA, headB
        if lenA > lenB:     # 让curB为最长链表的头，lenB为其长度
            curA, curB = curB, curA
            lenA, lenB = lenB, lenA 
        for _ in range(lenB - lenA):  # 让curA和curB在同一起点上（末尾位置对齐）
            curB = curB.next 
        while curA:         #  遍历curA 和 curB，遇到相同则直接返回
            if curA == curB:
                return curA
            else:
                curA = curA.next 
                curB = curB.next
        return None 

首先，计算两个链表的长度lenA和lenB，这一部分代码过于简单就不做赘述了。

        curA, curB = headA, headB
        if lenA > lenB:     # 让curB为最长链表的头，lenB为其长度
            curA, curB = curB, curA
            lenA, lenB = lenB, lenA 

将两个链表的头结点headA和headB分别赋值给curA和curB。

为了题目给定的两个链表不知道其长度，有可能是A长，也有可能是B长，因此正常逻辑应该是设计两个if语句，分析A长的情况和B长的情况，但这样过于繁琐。因此这里只设计一个if语句，针对A比B长的情况，将A和B的头结点和长度都进行互换，因此原始的长的A链表变为了现在的B链表，原始短的B链表变为了现在的A链表。你可以理解为不用考虑一开始题目给定的A链表和B链表，经过这步代码后，将长的那个链表定义为了B链表，而将短的那个链表定义为了A链表。

        for _ in range(lenB - lenA):  # 让curA和curB在同一起点上（末尾位置对齐）
            curB = curB.next 

利用for循环，将长的链表B的指针向后移动（lenB-lenA）个单位，这个意思是将长的链表B的指针从头结点向后移，直至后续长度与短的链表A的长度一样。

        while curA:         #  遍历curA 和 curB，遇到相同则直接返回
            if curA == curB:
                return curA
            else:
                curA = curA.next 
                curB = curB.next
        return None 

设计while循环，只要curA还存在，循环就一直存在。当curA和curB的值相等时，return到curA；不然的话将A链表和B链表的指针一格一格的向后移，直到找到相交点（curA）或者循环结束也没有相交点（return None）。