题目
输入两个链表,找出它们的第一个公共结点。
数据结构
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) :
val(x), next(NULL) {
}
};
分析
这道题目,我们从简单到复杂来看看,简单的情况下,如果两个链表等长,那么直接两个指针指向两个头结点,一个一个比下去就行了,可关键是两个链表并不一定等长,那么就得想办法来使得两个链表“等长”
一种解法是先遍历两个链表,得到两个链表的长度,从而得到长度差d,指针p指向长度较长的链表,指针q指向长度较短的链表,p先向后移动d个单位,这样p后面的链表和q后面的链表长度就等长了,之后的操作就是简单情况的操作了。
这里贴上代码
public ListNode FindFirstCommonNodeII(ListNode pHead1, ListNode pHead2) {
ListNode current1 = pHead1;// 链表1
ListNode current2 = pHead2;// 链表2
if (pHead1 == null || pHead2 == null)
return null;
int length1 = getLength(current1);
int length2 = getLength(current2);
// 两连表的长度差
// 如果链表1的长度大于链表2的长度
if (length1 >= length2) {
int len = length1 - length2;
// 先遍历链表1,遍历的长度就是两链表的长度差
while (len > 0) {
current1 = current1.next;
len--;
}
}
// 如果链表2的长度大于链表1的长度
else if (length1 < length2) {
int len = length2 - length1;
// 先遍历链表1,遍历的长度就是两链表的长度差
while (len > 0) {
current2 = current2.next;
len--;
}
}
//开始齐头并进,直到找到第一个公共结点
while(current1!=current2){
current1=current1.next;
current2=current2.next;
}
return current1;
}
// 求指定链表的长度
public static int getLength(ListNode pHead) {
int length = 0;
ListNode current = pHead;
while (current != null) {
length++;
current = current.next;
}
return length;
}
这里,我向大家介绍第二种解法,这种解法更加精炼优美
题目的关键在于两个链表长度不等长,上面的求长度差的方式是我们的第一种解决方案,其实我们也可以将链表1和链表2逻辑上组合在一起,一个变成链表1+链表2,一个变成链表2+链表1,这样就等效成了长度一样的情况
class Solution {
public:
ListNode* FindFirstCommonNode( ListNode* pHead1, ListNode* pHead2) {
ListNode* p1 = pHead1;
ListNode* p2 = pHead2;
while (p1!=p2){
// ?phead2是核心,相当于p1走到链表1的末尾后跳到了链表2的头部
p1 = (p1 == NULL) ? pHead2 : p1->next;
// 分析同上
p2 = (p2 == NULL) ? pHead1 : p2->next;
}
return p1;
}
};
好代码真的需要慢慢品