160. 相交链表

题目

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

在节点 c1 开始相交。

思路

• O(n)时间和O(1)空间。
• 先计算两个链表的长度，让长的链表先走几步，使得剩下的长度等于较短的链表。
• 然后两个链表同时同速度遍历，直到两个链表的节点相同。这里相同有两种情况，一种是找到了公共的节点，一种是到了结尾，都为nullptr。两种情况都可以直接返回使得循环停止的节点。

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    int cntLength(ListNode* head) {
        int cnt = 0;

        while ( head->next ) {
            cnt++;
            head = head->next;
        }

        return cnt;
    }

    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        if ( !headA || !headB )
            return nullptr;

        int ASize = cntLength(headA);
        int BSize = cntLength(headB);

        int diff = ASize - BSize;

        if ( ASize < BSize ) {
            diff *= -1;
            swap( headA, headB );
        }

        while ( diff-- ) 
            headA = headA->next;
        
        while ( headA != headB ) {
            headA = headA->next;
            headB = headB->next;
        }

        return headA;

    }

    void swap( ListNode*& a, ListNode*& b ) {
        ListNode* temp = a;
        a = b;
        b = temp;

        return;
    }
};