快慢指针在链表中的使用
链表毫无疑问是数据结构中最重要的知识点之一,因此许多面试官都很喜欢在链表上做文章。链表主要有这几个变种:单向链表,双向链表,循环链表。以链表为考点的面试题更是千变万化,令人眼花缭乱。
什么是单向链表?
单向链表(单链表)是最简单的一种链表,其特点是链表的链接方向是单向的,对链表的访问要通过顺序读取从头部开始;链表是使用指针进行构造的列表;又称为结点列表,因为链表是由一个个结点组装起来的;其中每个结点都有指针成员变量指向列表中的下一个结点;链表是由结点构成,head指针指向第一个成为表头结点,而终止于最后一个指向NULL的指针。
如图:
什么是双向链表?
双向链表也叫双链表,是链表的一种,它的每个数据结点中都有两个指针,分别指向直接后继和直接前驱。所以,从双向链表中的任意一个结点开始,都可以很方便地访问它的前驱结点和后继结点。一般我们都构造双向循环链表。
什么是循环链表?
循环链表是另一种形式的链式存贮结构。它的特点是表中最后一个结点的指针域指向头结点,整个链表形成一个环。
那么下面让我们来看一道十分简单的面试题:
给定一个带有头结点 head 的非空单链表,返回链表的中间结点。
如果有两个中间结点,则返回第二个中间结点。
示例1:
输入:[1,2,3,4,5]
输出:此列表中的结点 3 (序列化形式:[3,4,5])
返回的结点值为 3 。 (测评系统对该结点序列化表述是 [3,4,5])。
注意,我们返回了一个 ListNode 类型的对象 ans,这样:
ans.val = 3, ans.next.val = 4, ans.next.next.val = 5, 以及 ans.next.next.next = NULL.
示例2:
输入:[1,2,3,4,5,6]
输出:此列表中的结点 4 (序列化形式:[4,5,6])
由于该列表有两个中间结点,值分别为 3 和 4,我们返回第二个结点。
读完题目之后我们通常会想到的解法为:先遍历链表,得到其长度 l e n len len。然后再次遍历链表,此时位于 l e n 2 \frac{len}{2} 2len的元素cur,即是我们需要返回的中间结点。
class Solution
{
public:
ListNode* middleNode(ListNode* head)
{
int len = 0;
ListNode* cur = head;
while (cur != nullptr)
{
++len;
cur = cur -> next;
}
int k = 0;
cur = head;
while (k < n / 2)
{
++k;
cur = cur -> next;
}
return cur;
}
};
上面的代码在遍历时,为得到链表的长度执行了两次循环。但实际上很多时候,我们需要一步到位,仅仅凭借一次循环解决问题。并且真实的面试题通常也不会简单到通过一次遍历就让你知道整个链表的长度,当遇到循环链表的时候,我们这种以遍历求长度的方法就不再适用了。这时就需要用到快慢指针法。
在本题中,我们可以通过设计两个变量 s l o w , f a s t slow,fast slow,fast同时遍历链表。这样当 f a s t fast fast指向末尾时, s l o w slow slow就刚好在中间了,此时它代表的就是我们所需要返回的中间结点了。
class Solution
{
public:
ListNode* middleNode(ListNode* head)
{
ListNode* slow = head;
ListNode* fast = head;
while (fast != NULL && fast->next != NULL)
{
slow = slow -> next;
fast = fast -> next -> next;
}
return slow;
}
};
优化后的代码可读性更强,并且只需使用一次遍历。
快慢指针法在链表类面试题中的使用十分广泛,更多此类面试题请见《剑指Offer》。