剑指offer编程题--寻找两个链表公共结点

题目描述

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

思路

先分别计算两个链表的长度,然后对齐链表(就是把比较长的那个链表的指针先往前走几步,以便赶上短的那个),然后同步遍历两个链表,这样肯定会同时到达公共结点。

Java代码

public class Solution {
    public ListNode FindFirstCommonNode(ListNode pHead1, ListNode pHead2) {
         if(pHead1==null||pHead2==null){
             return null;
         }
        int size1 = 0;
        int size2 = 0;
        ListNode p1 = pHead1,p2=pHead2;
        //计算链表长度
        while(p1!=null){
            p1 = p1.next;
            size1++;
        }
        while(p2!=null){
            p2 = p2.next;
            size2++;
        }
        //对齐链表
        p1 = pHead1;p2=pHead2;
        int k = 0;
        if(size1>size2){
            k = size1-size2;
            while(k>0){
                p1 = p1.next;
                k--;
            }
        }else{
            k = size2-size1;
            while(k>0){
                p2 = p2.next;
                k--;
            }
        }
        //找公共节点
        while(p1!=p2){
            p1=p1.next;
            p2=p2.next;
        }
        if(p1!=null&&p2!=null){
            return p1;
        }else{
            return null;
        }

    }
}

C++代码

class Solution {
public:
    ListNode* FindFirstCommonNode( ListNode* pHead1, ListNode* pHead2) {
        if(pHead1 == NULL || pHead2 == NULL){
            return NULL;
        }
        int size1 = 0;
        int size2 = 0;
        ListNode* p1 = pHead1;
        ListNode* p2 = pHead2;
        //计算两个链表的长度
        while(p1!=NULL){
            p1 = p1->next;
            size1++;
        }
          while(p2!=NULL){
            p2 = p2->next;
            size2++;
        }
        p1 = pHead1;
        p2 = pHead2;
        //对齐链表
        int k  = 0;
        if(size1>size2){
            k = size1-size2;
            while(k>0){
                p1 = p1->next;
                k--;
            }
        }else if(size1<size2){
            k = size2-size1;
            while(k>0){
                p2 = p2->next;
                k--;
            }
        }
        //查找公共节点
        while(p1!=p2){
            p1=p1->next;
            p2=p2->next;
        }
        if(p1!=NULL&&p2!=NULL){
            return p1;
        }else{
            return NULL;
        }

    }
};
点赞