标签:
链表、Set
题目:
输入两个无环的单向链表,找出它们的第一个公共结点,如果没有公共节点则返回空。(注意因为传入数据是链表,所以错误测试数据的提示是用其他方式显示的,保证传入数据是正确的)
要求:空间复杂度 O(1)O(1),时间复杂度 O(n)O(n)
例如,输入{1,2,3},{4,5},{6,7}时,两个无环的单向链表的结构如下图所示:
可以看到它们的第一个公共结点的结点值为6,所以返回结点值为6的结点。
输入描述:
输入分为是3段,第一段是第一个链表的非公共部分,第二段是第二个链表的非公共部分,第三段是第一个链表和第二个链表的公共部分。 后台会将这3个参数组装为两个链表,并将这两个链表对应的头节点传入到函数FindFirstCommonNode里面,用户得到的输入只有pHead1和pHead2。
返回值描述:
返回传入的pHead1和pHead2的第一个公共结点,后台会打印以该节点为头节点的链表。
反思:
本题最关键的点在于链表中的公共节点,他们指向的内存地址空间是相同的,我之前用嵌套for循环想暴力破解的时候没有注意到这点,而是判断他们的值是否相等,导致了错误。
1、本道题目可以用集合HashSet的方式来解决,判断集合中有没有存在指向相同指针的元素。
2、嵌套for循环暴力破解。
用到的知识点:
暴力破解、HashSet集合
代码:
暴力破解:
public ListNode FindFirstCommonNode(ListNode pHead1, ListNode pHead2) {
ListNode sentinel = new ListNode(-1);
for (ListNode i = pHead1; i != null; i = i.next) {
for (ListNode j = pHead2; j != null ; j = j.next) {
if(i == j) {
sentinel.next = i;
return sentinel.next;
}
}
}
return sentinel.next;
}
HashSet集合解法
public ListNode FindFirstCommonNode(ListNode pHead1, ListNode pHead2) {
HashSet<ListNode> listNodeSet = new HashSet<>();
while(pHead1 != null) {
listNodeSet.add(pHead1);
pHead1 = pHead1.next;
}
while(pHead2 != null) {
if(listNodeSet.contains(pHead2)) {
return pHead2;
}else{
pHead2 = pHead2.next;
}
}
return null;
}