欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-07-03 19:18:19
...

题目:

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

我的解:

利用Set集合记录第一条链表遍历的所有节点,然后遍历第二条链表找到第二条链表中与第一条相同的节点。这个方法有点笨。

public class Solution {
    public ListNode FindFirstCommonNode(ListNode pHead1, ListNode pHead2) {
        Set<ListNode> set = new HashSet<ListNode>();
        ListNode p = pHead1;
        while(true){
            if(p == null) break;
            set.add(p);
            p = p.next;
        }
        p = pHead2;
        if(p == null) return null;
        while(!set.contains(p)){
            p = p.next;
            if(p == null) return null;
        }
        return p;
 
    }
}

更好的解:

用两个指针分别指向两个链表头,指针依次向下移动比对。
当两条链表一样长时在第一次遍历内就能找到公共节点。
如果两条链表不一样长,当指针走到链表尾部时跳到另一条链表的开头,这样最多遍历两次就能找到公共节点,代码很简短。
两条链表不一样长的情况下,当找到公共节点时,两个指针走了相同的路程。

class Solution {
public:
    ListNode* FindFirstCommonNode( ListNode *pHead1, ListNode *pHead2) {
        ListNode *p1 = pHead1;
        ListNode *p2 = pHead2;
        while(p1!=p2){
            p1 = (p1==NULL ? pHead2 : p1->next);
            p2 = (p2==NULL ? pHead1 : p2->next);
        }
        return p1;
    }
};

总结

做题前要多思考~~~