剑指Offer面试题:31.两个链表的第一个公共节点

时间:2021-08-22 11:02:46

一、题目:两个链表的第一个公共节点

题目:输入两个链表,找出它们的第一个公共结点。

  链表结点定义如下,这里使用C#语言描述:

    public class Node
{
public int key;
public Node nextNode;

public Node(int key)
{
this.key = key;
}
}

二、解题思路

2.1 蛮力法

  碰到这道题,很多人的第一反应就是蛮力法:在第一链表上顺序遍历每个结点,每遍历到一个结点的时候,在第二个链表上顺序遍历每个结点。如果在第二个链表上有一个结点和第一个链表上的结点一样,说明两个链表在这个结点上重合,于是就找到了它们的公共结点。如果第一个链表的长度为m,第二个链表的长度为n,显然该方法的时间复杂度是O(mn)

2.2 借助外部空间法

  首先,经过分析我们发现两个有公共结点而部分重合的链表,拓扑形状看起来像一个Y,而不可能像X,如下图所示,两个链表在值为6的结点处交汇:

剑指Offer面试题:31.两个链表的第一个公共节点

  如果两个链表有公共结点,那么公共结点出现在两个链表的尾部。如果我们从两个链表的尾部开始往前比较,最后一个相同的结点就是我们要找的结点。But,在单链表中只能从头结点开始按顺序遍历,最后才能到达尾结点。最后到达的尾结点却要最先被比较,这是“后进先出”的特性。于是,我们可以使用栈的特点来解决这个问题:分别把两个链表的结点放入两个栈里,这样两个链表的尾结点就位于两个栈的栈顶,接下来比较两个栈顶的结点是否相同。如果相同,则把栈顶弹出接着比较下一个栈顶,直到找到最后一个相同的结点

    public static Node FindFirstCommonNode(Node head1, Node head2)
{
if(head1 == null || head2 == null)
{
return null;
}

Stack
<Node> stack1 = new Stack<Node>();
Stack
<Node> stack2 = new Stack<Node>();

while(head1 != null)
{
stack1.Push(head1);
head1
= head1.nextNode;
}

while(head2 != null)
{
stack2.Push(head2);
head2
= head2.nextNode;
}

Node node1
= null;
Node node2
= null;
Node common
= null;

while(stack1.Count > 0 && stack2.Count > 0)
{
node1
= stack1.Peek();
node2
= stack2.Peek();

if (node1.key == node2.key)
{
common
= node1;
stack1.Pop();
stack2.Pop();
}
else
{
break;
}
}

return common;
}

  在上述思路中,我们需要用两个辅助栈。如果链表的长度分别为m和n,那么空间复杂度是O(m+n)。这种思路的时间复杂度也是O(m+n)。和最开始的蛮力法相比,时间效率得到了提高,相当于是用空间消耗换取了时间效率

2.3 不借助外部空间法

  那么,可不可以不借助栈来实现了呢?答案是可以的,我们可以首先遍历两个链表得到它们的长度,就能知道哪个链表比较长,以及长的链表比短的链表多几个结点。在第二次遍历的时候,在较长的链表上先走若干步,接着再同时在两个链表上遍历,找到的第一个相同的结点就是它们的第一个公共结点

  比如在上图的两个链表中,我们可以先遍历一次得到它们的长度分别为5和4,也就是较长的链表与较短的链表相比多一个结点。第二次先在长的链表上走1步,到达结点2。接下来分别从结点2和结点4出发同时遍历两个结点,直到找到它们第一个相同的结点6,这就是我们想要的结果。

    public static Node FindFirstCommonNode(Node head1, Node head2)
{
// 得到两个链表的长度
int length1 = GetListLength(head1);
int length2 = GetListLength(head2);
int diff = length1 - length2;

Node headLong
= head1;
Node headShort
= head2;
if (diff < 0)
{
headLong
= head2;
headShort
= head1;
diff
= length2 - length1;
}
// 先在长链表上走几步
for (int i = 0; i < diff; i++)
{
headLong
= headLong.nextNode;
}
// 再同时在两个链表上遍历
while (headLong != null && headShort != null && headLong != headShort)
{
headLong
= headLong.nextNode;
headShort
= headShort.nextNode;
}

Node commonNode
= headLong;
return commonNode;
}

private static int GetListLength(Node head)
{
int length = 0;
Node tempNode
= head;
while (tempNode != null)
{
tempNode
= tempNode.nextNode;
length
++;
}

return length;
}

  上述思路与借助栈的方法的时间复杂度都是O(m+n),但我们不再需要辅助的栈,因此提高了空间效率。

三、单元测试

3.1 测试用例:功能测试与特殊输入测试

    [TestClass]
public class CommonNodeHelperTest
{
private void DestoryNode(Node node)
{
if (node != null)
{
node
= null;
}
}

// 第一个公共结点在链表中间
// 1 - 2 - 3 \
// 6 - 7
// 4 - 5 /
[TestMethod]
public void FindTest1()
{
Node node1
= new Node(1);
Node node2
= new Node(2);
Node node3
= new Node(3);
Node node4
= new Node(4);
Node node5
= new Node(5);
Node node6
= new Node(6);
Node node7
= new Node(7);

// first
node1.nextNode = node2;
node2.nextNode
= node3;
node3.nextNode
= node6;
node6.nextNode
= node7;
// second
node4.nextNode = node5;
node5.nextNode
= node6;

Node actual
= CommonNodeHelper.FindFirstCommonNode(node1, node4);
Assert.AreEqual(actual.key,
6);

DestoryNode(node1);
DestoryNode(node2);
DestoryNode(node3);
DestoryNode(node4);
DestoryNode(node5);
DestoryNode(node6);
DestoryNode(node7);
}

// 没有公共结点
// 1 - 2 - 3 - 4
//
// 5 - 6 - 7
[TestMethod]
public void FindTest2()
{
Node node1
= new Node(1);
Node node2
= new Node(2);
Node node3
= new Node(3);
Node node4
= new Node(4);
Node node5
= new Node(5);
Node node6
= new Node(6);
Node node7
= new Node(7);

// first
node1.nextNode = node2;
node2.nextNode
= node3;
node3.nextNode
= node4;

// second
node5.nextNode = node6;
node6.nextNode
= node7;

Node actual
= CommonNodeHelper.FindFirstCommonNode(node1, node5);
Assert.AreEqual(actual,
null);

DestoryNode(node1);
DestoryNode(node2);
DestoryNode(node3);
DestoryNode(node4);
DestoryNode(node5);
DestoryNode(node6);
DestoryNode(node7);
}

// 公共结点是最后一个结点
// 5 - 6 \
// 7
// 1 - 2 - 3 - 4 /
[TestMethod]
public void FindTest3()
{
Node node1
= new Node(1);
Node node2
= new Node(2);
Node node3
= new Node(3);
Node node4
= new Node(4);
Node node5
= new Node(5);
Node node6
= new Node(6);
Node node7
= new Node(7);

// first
node1.nextNode = node2;
node2.nextNode
= node3;
node3.nextNode
= node4;
node4.nextNode
= node7;
// second
node5.nextNode = node6;
node6.nextNode
= node7;

Node actual
= CommonNodeHelper.FindFirstCommonNode(node5, node1);
Assert.AreEqual(actual.key,
7);

DestoryNode(node1);
DestoryNode(node2);
DestoryNode(node3);
DestoryNode(node4);
DestoryNode(node5);
DestoryNode(node6);
DestoryNode(node7);
}

// 公共结点是第一个结点
// 1 - 2 - 3 - 4 - 5
// 两个链表完全重合
[TestMethod]
public void FindTest4()
{
Node node1
= new Node(1);
Node node2
= new Node(2);
Node node3
= new Node(3);
Node node4
= new Node(4);
Node node5
= new Node(5);
Node node6
= new Node(6);
Node node7
= new Node(7);

// first & second
node1.nextNode = node2;
node2.nextNode
= node3;
node3.nextNode
= node4;
node4.nextNode
= node5;

Node actual
= CommonNodeHelper.FindFirstCommonNode(node1, node1);
Assert.AreEqual(actual.key,
1);

DestoryNode(node1);
DestoryNode(node2);
DestoryNode(node3);
DestoryNode(node4);
DestoryNode(node5);
DestoryNode(node6);
DestoryNode(node7);
}

// 输入的两个链表有一个空链表
[TestMethod]
public void FindTest5()
{
Node node1
= new Node(1);
Node node2
= new Node(2);
Node node3
= new Node(3);
Node node4
= new Node(4);
Node node5
= new Node(5);

// first & second
node1.nextNode = node2;
node2.nextNode
= node3;
node3.nextNode
= node4;
node4.nextNode
= node5;

Node actual
= CommonNodeHelper.FindFirstCommonNode(node1, null);
Assert.AreEqual(actual,
null);

DestoryNode(node1);
DestoryNode(node2);
DestoryNode(node3);
DestoryNode(node4);
DestoryNode(node5);
}

// 输入的两个链表均为空链表
[TestMethod]
public void FindTest6()
{
Node actual
= CommonNodeHelper.FindFirstCommonNode(null, null);
Assert.AreEqual(actual,
null);
}
}

3.2 测试结果:用例通过情况与代码覆盖率

  (1)用例通过情况

剑指Offer面试题:31.两个链表的第一个公共节点

  (2)代码覆盖率

剑指Offer面试题:31.两个链表的第一个公共节点

 

作者:周旭龙

出处:http://edisonchou.cnblogs.com

本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文链接。