相交链表

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

注意：

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

 

/**
 * 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) {
        if(headA == NULL || headB == NULL)
            return NULL;
        ListNode *p = headA;
        ListNode *q = headB;
        while(p->next != NULL){
            p = p->next;
        }
        while(q->next != NULL){
            q = q->next;
        }
        if(p != q){
            return NULL;
        }
        else{
            p = headA;
            q = headB;
            while(p != q){
                p = p->next;
                q = q->next;
                if(p == NULL)
                    p = headA;
                if(q == NULL)
                    q = headB;
            }
            return p;
        }
    }
};

以上是我的解法，这样虽然满足O(1)的空间复杂度，但是时间复杂度为O(mn)。

以下是官方给的更好的解法：时间复杂度为O(m+n)。