Little-Prince

导航

160. 相交链表

编写一个程序,找到两个单链表相交的起始节点。

如下面的两个链表

 

 

在节点 c1 开始相交。

注意:


    如果两个链表没有交点,返回 null.
    在返回结果后,两个链表仍须保持原有的结构。
    可假定整个链表结构中没有循环。
    程序尽量满足 O(n) 时间复杂度,且仅用 O(1) 内存。

思路:

双指针法

创建两个指针 pA 和 pB,分别初始化为链表 A 和 B 的头结点。然后让它们向后逐结点遍历。当 pA 到达链表的尾部时,将它重定位到链表 B 的头结点; 类似的,当 pB 到达链表的尾部时,将它重定位到链表 A 的头结点。若在某一时刻 pA 和 pB 相遇,则 pA/pB 为相交结点。pA 和 pB 两个指针走的长度都为 len_A + len_B。如果两个链表存在相交,它们必会走过一段相同的路程,也就是说他们的末尾的结点必然相同。因此当 pA/pB 到达链表结尾时,记录下链表 A/B 对应的元素。若最后元素不相同,则两个链表不相交。

代码:

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        
        ListNode *a = headA, *b = headB;
        if(!a||!b)
            return NULL;
        while(a||b)
        {
            if(a==b)
                return a;
            if(!a->next&&b->next)
            {
                a = headB;
                b = b->next;
            }
            else if(a->next&&!b->next)
            {
                a = a->next;
                b = headA;
            }
            else if(!a->next&&!b->next)
                return NULL;
            else
            {
                 a = a->next;
                 b = b->next;
            }
           
        }
        return NULL;
    }
};

 

posted on 2020-10-07 20:31  Little-Prince  阅读(155)  评论(0编辑  收藏  举报