题目:输入两个链表,找出它们的第一个公共结点。
面试的时候碰到这道题,很多应聘者的第一反应就是蛮力法:在第一链表上顺序遍历每个结点,没遍历到一个结点的时候,在第二个链表上顺序遍历每个结点。如果在第二个链表上有一个结点和第一个链表上的结点一样,说明两个链表在这个结点上重合,于是就找到了他们的公共结点。如果第一个链表的长度为m,第二个链表的长度为n,显然该方法的时间复杂度为O(mn).
通常蛮力法不会是最好的办法,我们接下来试着分析有公共结点的两个链表有哪些特点。从链表结点的定义可以看出,这两个链表是单链表。如果两个单向链表有公共的结点,那么这两个链表从某一结点开始,它们的m_pNext都指向同一个结点,但由于是单向链表的结点,之后他们所有结点都是重合的,不可能再出现分叉。所以两个有公共结点而部分重合的链表,拓扑形状开起来像一个Y,而不可能项X。
经过分析我们发现,如果两个链表有公共结点,那么公共结点出现在两个链表的尾部。如果我们从两个链表的尾部开始往前比较,最后一个相同的结点就是我们要找的结点。可问题是在单向链表中,我们只能从头结点开始按顺序遍历,最后才能到达尾节点。最后到达的尾节点却要最先被比较,这听起来是不是像后进先出,也是我们就能想到用栈的特点来解决这个问题:分别把两个链表的结点放入两个栈里,这样两个链表的尾节点就位于两个栈的栈顶,接下来比较两个栈顶的结点是否相同。如果相同,则把栈顶弹出接着比较下一个栈顶,直到找到最后一个相同的结点。
在上述思路中,我们需要用两个辅助栈。如果链表的长度分别为m和n,那么空间复杂度是O(m+n)。这种思路的时间复杂度也是O(m+n).和最开始的蛮力法相比,时间效率得到了提高,相当于是用空间消耗换取了时间效率。
之所以需要用到栈,是因为我们想同时遍历到达两个栈的尾节点。当两个链表的长度不相同时,如果我们从头开始遍历到达两个栈的尾节点的时间就不一致。其实解决这个问题还有一个更简单的办法:首先遍历两个链表得到他们的长度,就能知道哪个链表比较长,以及长的链表比短的链表多几个结点。在第二次遍历的时候,在较长的链表上先走若干步,接着再同时在两个链表上遍历,找到第一个相同的结点就是他们的第一个公共结点。
第三种思路比第二种思路相比,时间复杂度为O(m+n),但我们不在需要辅助栈,因此提高了空间效率。当面试官肯定了我们的最后一个思路的时候,可以动手写代码了。
代码:
public class Offer37 {
public static class ListNode{
int value;
ListNode next;
ListNode(){}
ListNode(int val){
this.value = val;
}
public int getValue(){
return value;
}
@Override
public String toString(){
StringBuilder sb = new StringBuilder();
sb.append(" value= "+value);
sb.append(" next= "+(next == null?"null" : next.value));
return sb.toString();
}
}
public ListNode findFirstCommonNode(ListNode head1, ListNode head2){
if(head1 == null || head2== null){
System.out.print("invalid input ");
return null;
}
int length1 = getLength(head1);
int length2 = getLength(head2);
int lenDiff = length1 - length2;
ListNode headLong = head1;
ListNode headShort = head2;
if(length2 > length1){
headLong = head2;
headShort = head1;
lenDiff = length2-length1;
}
//先在长链表上走积不,再同时在两个链表上遍历
for(int i = 0;i<lenDiff;i++)
headLong = headLong.next;
while( headLong != headShort){
//书上是headLong != null && headShort!=null &&headLong != headShort,结果headShortweinull,输出空指针异常
headLong = headLong.next;
headShort = headShort.next;
}
//int v = headLong.value;
return headShort;
}
public int getLength(ListNode head) {
int result = 0;
if(head == null)
return result;
ListNode point = head;
while(point != null){
point = point.next;
result++;
}
return result;
}
public static void printLinkedList(ListNode head){
while(head != null){
System.out.print(head.value+"->");
head = head.next;
}
System.out.println("null");
}
public static void main(String[] args) {
Offer37 of37 = new Offer37();
//功能测试,1,输入的两个链表有公共节点:第一个公共节点在链表的中间
// 第一个公共结点在链表中间
// 1 - 2 - 3 \
// 6 - 7
// 4 - 5 /
ListNode node11a = new ListNode(1);
ListNode node12a = new ListNode(2);
ListNode node13a = new ListNode(3);
ListNode node14b = new ListNode(4);
ListNode node15b = new ListNode(5);
ListNode node16a = new ListNode(6);
ListNode node17a = new ListNode(7);
node11a.next = node12a;
node12a.next = node13a;
node13a.next = node16a;
node16a.next = node17a;
node14b.next = node15b;
node15b.next = node16a;
printLinkedList(node11a);
printLinkedList(of37.findFirstCommonNode(node11a, node14b));
//功能测试,2,输入的两个链表有公共节点: 公共结点是最后一个结点
// 1 - 2 - 3 - 4 \
// 7
// 5 - 6 /
ListNode node21a = new ListNode(1);
ListNode node22a = new ListNode(2);
ListNode node23a = new ListNode(3);
ListNode node24a = new ListNode(4);
ListNode node25b = new ListNode(5);
ListNode node26b = new ListNode(6);
ListNode node27a = new ListNode(7);
node21a.next = node22a;
node22a.next = node23a;
node23a.next = node24a;
node25b.next = node26b;
node24a.next = node27a;
node26b.next = node27a;
printLinkedList(of37.findFirstCommonNode(node21a, node25b));
//功能测试,3,输入的两个链表有公共节点: 公共结点是第一个结点
// 1 - 2 - 3 - 4 - 5
// 两个链表完全重合
ListNode node31a = new ListNode(1);
ListNode node32a = new ListNode(2);
ListNode node33a = new ListNode(3);
ListNode node34a = new ListNode(4);
ListNode node35a = new ListNode(5);
node31a.next = node32a;
node32a.next = node33a;
node33a.next = node34a;
node34a.next = node35a;
printLinkedList(of37.findFirstCommonNode(node31a, node31a));
//功能测试,4, 没有公共结点
// 1 - 2 - 3 - 4
//
// 5 - 6 - 7
ListNode node41a = new ListNode(1);
ListNode node42a = new ListNode(2);
ListNode node43a = new ListNode(3);
ListNode node44a = new ListNode(4);
ListNode node45b = new ListNode(5);
ListNode node46b = new ListNode(6);
ListNode node47b = new ListNode(7);
node41a.next = node42a;
node42a.next = node43a;
node43a.next = node44a;
node45b.next = node46b;
node46b.next = node47b;
printLinkedList(of37.findFirstCommonNode(node41a, node45b));
//特殊输入测试,输入的链表头指针为NULL
ListNode node51a = null;
ListNode node51b = null;
printLinkedList(of37.findFirstCommonNode(node51a, node51b));
}
}
运行结果:
1->2->3->6->7->null
6->7->null
7->null
1->2->3->4->5->null
null
invalid input null