LeetCode 160.相交链表 (C++)

题目地址：力扣

这道题和之前刷过的一道题LeetCode 141. 环形链表(C++)很类似，两者都可以用相同的方法解

解法1：记录下HeadA走过的所有节点，然后HeadB去匹配看看有没有

class Solution {
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        set<ListNode *> node_set;
        while (headA != nullptr)    // headA的每一个节点都记录下来
        {
            node_set.insert(headA);
            headA = headA->next;
        }
        while (headB != nullptr)
        {
            if (node_set.find(headB) == node_set.end())    // 如果无匹配就往下走
                headB = headB->next;
            else    // 匹配上了就返回
                return headB;
        }    // 走完了都没匹配就说明没有重复的地方
        return nullptr;
    }
};

解法2：双指针

这种方法也很巧妙，有一个视频可以完全诠释这种方法：双指针解法

大致思路就是：当第一个节点走到第一条链尾的时候，将节点放在第二条链头，然后继续走；当第二个节点走到第二条链尾的时候，将节点放在第一条链头，然后继续走。每次这两个节点都只走一步，如果他们之间有重合的部分，那么必然会在第一个地方相遇。如果没有重合的部分，那么最后就会在nullptr处相遇。

假设第一条链长度为a，第二条链长度为b，长度不包括重合部分。若重合部分链长为c；那么用着个方法，节点在相遇（或同指向nullptr的时候），第一条链最后走的总长度就是a+c+b，第二条链走的总长度就是b+c+a。因此这样如果有重合部分，两个节点必然相遇。

class Solution {
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        if (headB == nullptr) // 题目中说了A链长度至少为1
            return nullptr;

        ListNode* nodeA = headA;
        ListNode* nodeB = headB;
        while (nodeA != nodeB) // 两个节点只要不相同就继续走
        {
            // 走完自己的链就去走对方的链
            nodeA = nodeA != nullptr ? nodeA->next : headB;    
            nodeB = nodeB != nullptr ? nodeB->next : headA;
        }
        return nodeA;
        }
};

Accepted

• 39/39 cases passed (40 ms)
• Your runtime beats 63.23 % of cpp submissions
• Your memory usage beats 64.63 % of cpp submissions (14.2 MB)