面试题 02.07. 链表相交

最新推荐文章于 2022-10-21 00:09:11 发布

Isabelle_Yan

最新推荐文章于 2022-10-21 00:09:11 发布

阅读量62

点赞数

分类专栏：代码随想录文章标签：链表数据结构

本文链接：https://blog.csdn.net/sinat_27523689/article/details/126512268

版权

代码随想录专栏收录该内容

14 篇文章 0 订阅

订阅专栏

面试题 02.07. 链表相交

题目链接

思路

设立两个指针，两个循序嵌套遍历，遇到相同的节点就返回。全部遍历结束后返回空。

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        ListNode *l1=headA;
        ListNode *l2 = headB;
        while(l1){
            l2 = headB;
            while(l2){
                if(l1==l2) return l2;
                else l2 = l2->next;
            }
            l1 = l1->next;
        }
        return NULL;
    }
};

代码随想录思路

在这里插入图片描述
我们求出两个链表的长度，并求出两个链表长度的差值，然后让curA移动到，和curB 末尾对齐的位置，如图：

在这里插入图片描述

此时我们就可以比较curA和curB是否相同，如果不相同，同时向后移动curA和curB，如果遇到curA == curB，则找到交点。

否则循环退出返回空指针。

class Solution {
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        ListNode* curA = headA;
        ListNode* curB = headB;
        int lenA = 0, lenB = 0;
        while (curA != NULL) { // 求链表A的长度
            lenA++;
            curA = curA->next;
        }
        while (curB != NULL) { // 求链表B的长度
            lenB++;
            curB = curB->next;
        }
        curA = headA;
        curB = headB;
        // 让curA为最长链表的头，lenA为其长度
        if (lenB > lenA) {
            swap (lenA, lenB);
            swap (curA, curB);
        }
        // 求长度差
        int gap = lenA - lenB;
        // 让curA和curB在同一起点上（末尾位置对齐）
        while (gap--) {
            curA = curA->next;
        }
        // 遍历curA 和 curB，遇到相同则直接返回
        while (curA != NULL) {
            if (curA == curB) {
                return curA;
            }
            curA = curA->next;
            curB = curB->next;
        }
        return NULL;
    }
};