算法学习12：两个单链表相交的一系列问题

题目

在本题中，单链表可能有环，也可能无环。给定两个单链表的头节点 head1和head2，这两个链表可能相交，也可能不相交。请实现一个函数， 如果两个链表相交，请返回相交的第一个节点；如果不相交，返回null 即可。 要求：如果链表1的长度为N，链表2的长度为M，时间复杂度请达到 O(N+M)，额外空间复杂度请达到O(1)。

分析

首先要判断这两个单链表是否有环。因为有环的情况下需要单独讨论。若两个链表都没环，有可能相交也有可能不相交，相交时，末节点必定为同一个。若两个都有环，则需要分三种情况讨论，不相交，有共同的入环节点 不同的入环节点。

代码

首先判断单链表是否有环，若有环返回入环节点，若无环，返回null。
先让快慢指针在环中相遇，若某个节点的指针域为空，必然无环，返回null。若不存在节点指针域为null，则必然存在一个节点使得快慢指针在此相遇，此时，将快指针指向链表头结点，快慢指针同时每次移动一个节点，当两个指针再次相遇，该节点就是入环节点。

ListNode* getLoopNode(ListNode* head) {
    if (head==NULL||head->next == NULL || head->next->next == NULL) {
        return NULL;
    }
    ListNode* fast = head->next->next;
    ListNode* slow = head->next;
    while (fast != slow) {
        if (fast->next == NULL || fast->next->next) {
            return NULL;
        }
        fast = fast->next->next;
        slow = slow->next;
    }
    fast = head;
    while (fast != slow) {
        fast = fast->next;
        slow = slow->next;
    }
    return fast;
}
ListNode* noLoop(ListNode* head1, ListNode* head2) {
    if (head1 == NULL || head2 == NULL) {
        return NULL;
    }
    int n = 0;
    ListNode* cur1 = head1;
    ListNode* cur2 = head2;
    while (cur1->next != NULL) {
        n++;
        cur1 = cur1->next;
    }
    while (cur2->next != NULL) {
        n--;
        cur2 = cur2->next;
    }
    cur1 = n > 0 ? head1 : head2;
    cur2 = cur1 == head1 ? head2 : head1;
    int n = abs(n);
    while (n--) {
        cur1 = cur1->next;

    }
    while (cur1 != cur2) {
        cur1 = cur1->next;
        cur2 = cur2->next;
    }
    return cur1;
}
ListNode* bothLoop(ListNode* head1, ListNode* loop1, ListNode* head2, ListNode* loop2) {
    ListNode* cur1 = NULL;
    ListNode* cur2 = NULL;
    if (loop1 == loop2) {
        cur1 = head1;
        cur2 = head2;
        int n = 0;
        while (cur1 != loop1) {
            n++;
            cur1 = cur1->next;
        }
        while (cur2 != loop2) {
            n--;
            cur2 = cur2->next;
        }
        cur1 = n > 0 ? head1 : head2;
        cur2 = cur1 == head1 ? head2 : head1;
        int n = abs(n);
        while (n--) {
            cur1 = cur1->next;
        }
        while (cur1 != cur2) {
            cur1 = cur1->next;
            cur2 = cur2->next;
        }
        return cur1;
    }
    else {
        cur1 = cur1->next;
        while (cur1 != loop1) {
            if (cur1 == loop2) {
                return loop2;
            }
            cur1 = cur1->next;
        }
        return NULL;
    }
}
ListNode* getIntersecNode(ListNode* head1, ListNode* head2) {
    if (head1 == nullptr || head2 == NULL) {
        return NULL;
    }
    ListNode* loop1 = getLoopNode(head1);
    ListNode* loop2 = getLoopNode(head2);
    if (loop1 == NULL && loop2 == NULL) {
        return noLoop(head1, head2);
    }
    if (loop1 != NULL && loop2 != NULL) {
        return bothLoop(head1, loop1, head2, loop2);
    }
    return NULL;
}