面试题22 链表中倒数第k个节点 (Python3) 单指针/双指针

输入一个链表，输出该链表中倒数第k个节点。为了符合大多数人的习惯，本题从1开始计数，即链表的尾节点是倒数第1个节点。例如，一个链表有6个节点，从头节点开始，它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。

示例：

给定一个链表: 1->2->3->4->5, 和 k = 2.

返回链表 4->5.

思路1 单指针

思路

最naive的思路可能就是先遍历一遍列表得到链表的长度N;

之后再重新从第一个节点开始遍历，返回第(N-k)个节点即可。

 

代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
        # 得到长度
        cur = head
        length = 0
        while cur:
            length = length+1
            cur = cur.next
        
        # 求出倒数第k个节点
        cur = head
        for i in range(length-k):
            cur = cur.next
        
        return  cur

结果

复杂度

时间复杂度O(2N)

空间复杂度O(1)

 

 

思路2 双指针

思路1需要先完整遍历一遍链表之后再从头遍历，显然有些耗时。

思路

使用双指针可以不需要第一次的完整遍历，思路如下：

首先再head处设置两个指针，分别是front，end；

之后front向前next k次，end不动，使得在以front为终点的链表里，end恰好为第倒数k个节点；

继续同时将front和end向前next，直到front到达终点，成为真正的尾指针；

此时返回end即可。

 

代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
        front = head
        end  = head
        # 拉开end与front之间的距离至k
        for i in range(k):
            front = front.next
        
        while front:
            front, end = front.next, end.next

        return end

结果

复杂度

时间复杂度O(N)

空间复杂度O(1)

 

总结

通过双指针之间的距离实现倒数第k个节点的思路不容易想到。