目录
牛客_两个链表的第一个公共结点
输入两个无环的单向链表,找出它们的第一个公共结点,如果没有公共节点则返回空。(注意因为传入数据是链表,所以错误测试数据的提示是用其他方式显示的,保证传入数据是正确的)
数据范围: n≤1000
要求:空间复杂度 O(1),时间复杂度 O(n)
例如,输入{1,2,3},{4,5},{6,7}时,两个无环的单向链表的结构如下图所示:
可以看到它们的第一个公共结点的结点值为6,所以返回结点值为6的结点。
输入描述:
输入分为是3段,第一段是第一个链表的非公共部分,第二段是第二个链表的非公共部分,第三段是第一个链表和第二个链表的公共部分。 后台会将这3个参数组装为两个链表,并将这两个链表对应的头节点传入到函数FindFirstCommonNode里面,用户得到的输入只有pHead1和pHead2。
返回值描述:
返回传入的pHead1和pHead2的第一个公共结点,后台会打印以该节点为头节点的链表。
题目解析
根据两个链表走的路程相同,找到相交点。
C++代码1
(双指针_长的先走)
/*
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) :
val(x), next(NULL) {
}
};*/
class Solution {
public:
ListNode* FindFirstCommonNode( ListNode* pHead1, ListNode* pHead2) {
ListNode* cur1 = pHead1;
int cnt1 = 0, cnt2 = 0;
while(cur1 != nullptr)
{
cnt1++;
cur1 = cur1->next;
}
ListNode* cur2 = pHead2;
while(cur2 != nullptr)
{
cnt2++;
cur2 = cur2->next;
}
// cout << pHead1->val << " " << pHead2->val << endl;
if(cnt1 < cnt2)
swap(pHead1, pHead2);
cur1 = pHead1, cur2 = pHead2;
// cout << pHead1->val << " " << pHead2->val << endl;
int diff = abs(cnt1 - cnt2);
while(diff--)
{
cur1 = cur1->next;
}
while(cur1 != nullptr && cur2 != nullptr)
{
// cout <<cur1->val << " " << cur2->val << endl;
if(cur1->val == cur2->val)
return cur1;
cur1 = cur1->next;
cur2 = cur2->next;
}
return nullptr;
}
};
C++代码2
(等量关系)
/*
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) :
val(x), next(NULL) {
}
};*/
class Solution {
public:
ListNode* FindFirstCommonNode( ListNode* pHead1, ListNode* pHead2) {
auto cur1 = pHead1, cur2 = pHead2; // 解法二-等量关系
while(cur1 != cur2)
{
cur1 = cur1 != nullptr ? cur1->next : pHead2; // 为空就到另个链表重新走
cur2 = cur2 != nullptr ? cur2->next : pHead1;
}
return cur1;
}
};
Java代码
import java.util.*;
/*
public class ListNode {
int val;
ListNode next = null;
ListNode(int val) {
this.val = val;
}
}*/
public class Solution
{
public ListNode FindFirstCommonNode(ListNode pHead1, ListNode pHead2)
{
ListNode cur1 = pHead1, cur2 = pHead2;
while(cur1 != cur2)
{
cur1 = cur1 != null ? cur1.next : pHead2;
cur2 = cur2 != null ? cur2.next : pHead1;
}
return cur1;
}
}