题目地址:力扣
这道题和之前刷过的一道题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)