问题描述
来自牛客剑指offer
输入两个链表,找出它们的第一个公共结点。(注意因为传入数据是链表,所以错误测试数据的提示是用其他方式显示的,保证传入数据是正确的)
思路
- 首先链表相交的结构是这样的:(图片来自牛客官方):
- 问题可以分为两种情况:
- 等长:那就很简单了,从头到尾遍历,找到相同的就返回节点,找不到就返回null。
- 不等长:这也是问题的比较值得思考的地方。如上图所示,不等长情况下如果直接从头遍历一次是一定找不到的。
- 对于不等长的简单思路就是,2层循环遍历,复杂度为O(n*m),最好想的方法,但是不太好。
- 我们发现,等长情况下时间复杂度降低为O(n),很高的效率,那么我们是否可以将问题转化成等长的链表呢?
- 那么我们可以比较 两个链表的长度,找出差值之后(比如A比B多一个节点),那么我们就让长的链表的指针往后走一个之后,再开始遍历,这样就是等长的情况遍历了。我们发现从公共节点后,链表是共享的,所以不可能说这种提前移动的操作会错过公共节点。
- 在讨论区发现了一种极其牛逼的方法,利用链表的循环来定位开始的位置,这样代码极为简洁,但是会改变链表的结构,使用要视情况而定**。读者可以试着画一个循环定位的过程就会理解其正确性**。
Code
public ListNode FindFirstCommonNode(ListNode pHead1, ListNode pHead2) {
ListNode l1 = pHead1;
ListNode l2 = pHead2;
int count1 = 0;
int count2 = 0;
while (l1 != null || l2 != null) {
if (l1 != null) {
count1++;
l1 = l1.next;
}
if (l2 != null) {
count2++;
l2 = l2.next;
}
}
l1 = pHead1;
l2 = pHead2;
if (count1 > count2) {
int div = count1 - count2;
while (div > 0) {
l1 = l1.next;
div--;
}
}
else {
int div = count2 - count1;
while (div > 0) {
l2 = l2.next;
div--;
}
}
while (l1 != null && l2 != null) {
if (l1 == l2) return l1;
l1 = l1.next;
l2 = l2.next;
}
return null;
}
// 循环链表定位的代码
public ListNode FindFirstCommonNode(ListNode pHead1, ListNode pHead2) {
if (pHead1 == null || pHead2 == null) return null;
ListNode p1 = pHead1;
ListNode p2 = pHead2;
while (p1 != p2) {
p1 = p1 == null ? pHead2 : p1.next;
p2 = p2 == null ? pHead1 : p2.next;
}
return p1;
}