剑指Offer之两个链表的第一个公共节点

版权声明:所有的博客仅仅作为个人笔记使用!!!!!!! https://blog.csdn.net/qq_35976351/article/details/83014398

题目描述

输入两个链表,找出它们的第一个公共结点。

解题思路

首先,如果都是空链表,那么肯定返回空。两个单链表假设有公共节点,那么肯定是“Y”字型的。

  • 先计算两个链表的长度,求出长度差
  • 根据长度差,后移长链表的头指针,使得两个链表的现在的头指针的位置距离为节点一样
  • 同步移动,直到指针相等,是Y型的连接处,也就是公共点;注意没有公共点的情况。

AC代码

class Solution {
  public:
    ListNode* FindFirstCommonNode( ListNode* pHead1, ListNode* pHead2) {
        if(pHead1 == nullptr || pHead2 == nullptr) {  // 日常判空
            return nullptr;
        }
        int len1 = 0, len2 = 0;
        // 计算链1的长度
        ListNode* p = pHead1;
        while(p != nullptr) {
            ++len1;
            p = p->next;
        }
        // 计算链2的长度
        p = pHead2;
        while(p != nullptr) {
            ++len2;
            p = p->next;
        }
        // p1是长链,p2是短链
        auto p1 = (len1 > len2) ? pHead1 : pHead2;
        auto p2 = (len1 <= len2) ? pHead1 : pHead2;
        int len = max(len1, len2) - min(len1, len2);
        while(len > 0) {  // 长链后移,同步长度
            p1 = p1->next;
            --len;
        }
        // 寻找公共节点,注意没有的情况
        while(p1 != p2 && p1 != nullptr) {
            p1 = p1->next;
            p2 = p2->next;
        }
        return p1;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_35976351/article/details/83014398