算法打卡day18-两个链表的第一个公共结点

79 阅读2分钟

描述

输入两个无环的单向链表,找出它们的第一个公共结点,如果没有公共节点则返回空。(注意因为传入数据是链表,所以错误测试数据的提示是用其他方式显示的,保证传入数据是正确的)

数据范围: n≤1000
要求:空间复杂度O(1),时间复杂度O(n)

例如,输入{1,2,3},{4,5},{6,7}时,两个无环的单向链表的结构如下图所示:

可以看到它们的第一个公共结点的结点值为6,所以返回结点值为6的结点。

输入描述:

输入分为是3段,第一段是第一个链表的非公共部分,第二段是第二个链表的非公共部分,第三段是第一个链表和第二个链表的公共部分。 后台会将这3个参数组装为两个链表,并将这两个链表对应的头节点传入到函数FindFirstCommonNode里面,用户得到的输入只有pHead1和pHead2。

返回值描述:

返回传入的pHead1和pHead2的第一个公共结点,后台会打印以该节点为头节点的链表。

示例1

输入:

{1,2,3},{4,5},{6,7}

返回值:

{6,7}

说明:

第一个参数{1,2,3}代表是第一个链表非公共部分,第二个参数{4,5}代表是第二个链表非公共部分,最后的{6,7}表示的是2个链表的公共部分
这3个参数最后在后台会组装成为2个两个无环的单链表,且是有公共节点的          

示例2

输入:

{1},{2,3},{}

返回值:

{}

说明:

2个链表没有公共节点 ,返回null,后台打印{}

代码实现

//遍历两表求差值,谁更长谁先走,走过差值一起走,相遇即是第一公共点。
struct ListNode* FindFirstCommonNode(struct ListNode* pHead1,
                                     struct ListNode* pHead2 ) {
    int len1 = 0, len2 = 0;
    struct ListNode* p1 = pHead1;
    struct ListNode* p2 = pHead2;
    for (; p1 != NULL; len1++)    p1 = p1->next;  //求len1
    for (; p2 != NULL; len2++)    p2 = p2->next;  //求len2
    if (len1 == 0 || len2 == 0)
        return NULL;   //某一个链表为空,肯定没公共点,两个都空的话,返回null
    else if (len1 >= len2) {   //若链表1更长
        int dis = len1 - len2;   //求链表长度差
        while (dis > 0) {
            pHead1 = pHead1->next;
            dis--;
        }      //长表先走,走到从该点开始的长度与短表相同时停下
    } else {         //链表2更长,情况同上
        int dis = len2 - len1;
        while (dis > 0) {
            pHead2 = pHead2->next;
            dis--;
        }
    }       //长表先走,走到从该点开始的长度与短表相同时停下
    while (pHead1 != pHead2) {   //两表一起走,相遇时停下
        pHead1 = pHead1->next;
        pHead2 = pHead2->next;
    }
    return pHead1;        //当然,return pHead2也可以
}