剑指offer52:两个链表的第一个公共节点

 

目录

1.暴力

2.看起来高级一点的做法

---

1.暴力

我们可以分别算出两个链表的长度，然后将长链表的指针指向和短指针距离尾部相同距离的位置，两个指针同时开始往后移动，看是否移到相同的节点

代码如下：

class Solution {
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        int len_a,len_b;
        len_a=len_b=0;
        ListNode *a=headA,*b=headB;
        while(a!=NULL)
        {
            len_a++;
            a=a->next;
        }
        while(b!=NULL)
        {
            len_b++;
            b=b->next;
        }
        a=headA;
        b=headB;
        while(len_a>len_b)
        {
            len_a--;
            a=a->next;
        }
        while(len_a<len_b)
        {
            len_b--;
            b=b->next;
        }
        while(a!=NULL && b!=NULL)
        {
            if(a==b)
            {
                return a
            }
            a=a->next;
            b=b->next;
        }
        return NULL;
    }
};

结果是这样：

2.看起来高级一点的做法

对上面第一个例子，我们可以设置两个指针，分别指向两个头，一起向后移动，将第一个到头的指针指向另外一个链表头部，画一画，其实这样就统一了步调，代码如下所示：

class Solution {
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        ListNode *ptrA=headA,*ptrB=headB;
        while(ptrA!=ptrB)
        {
            ptrA=(ptrA==NULL)?headB:ptrA->next;
            ptrB=(ptrB==NULL)?headA:ptrB->next;
        }
        if(ptrA==ptrB) return ptrA;
        return NULL;
    }
};

是不是少了很多代码看着很舒服？嘿嘿~

之前有一次写成这样了：

class Solution {
public:
    ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
        ListNode *ptrA=headA,*ptrB=headB;
        while(ptrA!=ptrB)
        {
            ptrA=(ptrA->next==NULL)?headB:ptrA->next;
            ptrB=(ptrB->next==NULL)?headA:ptrB->next;
        }
        if(ptrA==ptrB) return ptrA;
        return NULL;
    }
};

这里的问题在于，如果两个链表没有相交，但是ptrA和ptrB永远不会变成NULL，所以会无限循环。。。。。。

有一样错误的同学可以一步一步试着做做看...