力扣面试题：链表相交

题目描述

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

题目数据 保证 整个链式结构中不存在环。
注意，函数返回结果后，链表必须 保持其原始结构 。

举例

• 例1
  

  输入：intersectVal = 8, listA = [4,1,8,4,5], listB = [5,0,1,8,4,5], skipA = 2, skipB = 3
  输出：Intersected at '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
  输出：Intersected at '2'
  解释：相交节点的值为 2 （注意，如果两个链表相交则不能为 0）。
  从各自的表头开始算起，链表 A 为 [0,9,1,2,4]，链表 B 为 [3,2,4]。
  在 A 中，相交节点前有 3 个节点；在 B 中，相交节点前有 1 个节点。
  

解题思路

简单来说，就是求两个链表交点节点的指针。 注意，交点不是数值相等，而是指针相等。

首先定义两个指针分别指向两个链表的头结点，然后求两个链表的长度
计算两个链表的差
将长的链表，让指针先右移 差值 个位置
然后两个指针同时右移（直到链表结束），同时判断指针是否相等，
如果相等，返回当前结点，不相等，指针后移，直到链表结束
如果不相交，则返回 NULL

代码

（C++）

struct ListNode{
	int val;
	ListNode* next;
	ListNode(): val(0), next(nullptr) {}
	ListNode(int x): val(x), next(nullptr) {}
	ListNode(int x, ListNode* next): val(x), next(next) {}
};
class Solution {
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
		ListNode* curA = headA;
		ListNode* curB = headB;
		// 计算链表的长度
		int lenA = 0, lenB = 0;
		while (curA) {
			lenA++;
			curA = curA->next;
		}
		while (curB) {
			lenB++;
			curB = curB->next;
		}
		// 让指针重新指向头节点
		curA = headA;
		curB = headB;
		// 始终让 curA 指向长的链表
		if (lenB > lenA) {
			swap(lenA, lenB);
			swap(curA, curB);
		}
		// 让两个链表 尾对齐，curA右移
		int gap = lenA - lenB;
		while (gap--) {
			curA = curA->next;
		}
		while (curA) {
			if (curA == curB) {
				return curA;
			} else {
				curA = curA->next;
				curB = curB->next;
			}
		}
		return NULL;
	}
};

（Python）

class Node():
	def __init__(self, val):
		self.val = val;
		self.next = None
class Solution:
    def getIntersectionNode(self, headA: ListNode, headB: ListNode) -> ListNode:
        curA = headA
        curB = headB

        lenA = lenB = 0

        while (curA):
            lenA += 1
            curA = curA.next
        while (curB):
            lenB += 1
            curB = curB.next
        
        curA = headA
        curB = headB
        
        if (lenB > lenA):
            lenA, lenB = lenB, lenA
            curA, curB = curB, curA

        gap = lenA - lenB
        while gap:
            gap -= 1
            curA = curA.next
        
        while curA:
            if curA == curB:
                return curA
            else:
                curA = curA.next
                curB = curB.next
        return None