输入两个链表,找出它们的第一个公共结点
程序员文章站
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;
}
};
总结
做题前要多思考~~~
上一篇: Java三种方式交换两个变量的值
下一篇: Attributes in Clang