Leetcode 面试题02.07. 链表相交
题目描述
给你两个单链表的头节点 headA
和 headB
,请你找出并返回两个单链表相交的起始节点。如果两个链表没有交点,返回 null
。
题解
这个题比较自然的想法先求出链表A和B的长度之差,将之对其后一起往前走,若相交则有交点,否则无交点。弊端是处理方式太复杂,需要判断谁长谁短以及对齐。
有没有比较简单的解法呢? 有的有的
如果他们有相交节点,设相交段长度为len1,A链表中除了相交段的长度为len2,B链表中除了相交段的长度为len3,则有len2+len1+len3=len3+len1+len2,好像是句废话,从这个式子来看,如果他们有交点的话,从各自起点出发走这些路程最后一定能走到交点。因此对于每一个链表,只需要遍历到结尾的时候换对方的链表继续走就可以,最后若有相交节点一定会同时到达相交节点。
class Solution {
public:
ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
if(headA == nullptr || headB == nullptr) return nullptr;
ListNode *p = headA;
ListNode *q = headB;
while(p != q) {
if(p) p = p -> next;
else p = headB;
if(q) q = q -> next;
else q = headA;
}
return p;
}
};