一、题目
输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。
示例:
给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.
二、思路及代码实现
思路:双指针
由于单链表只有从前往后的指针,所以我们只能从头节点往后遍历。假设链表有 n 个节点,那么倒数第 k 个节点就是正数的第 n - k + 1 个节点。所以我们要先遍历一次链表,统计链表的节点数 n,再遍历一次链表,从头节点往后走 n - k + 1 步就能找到倒数第 k 个节点。这种方法的实现很简单,但是需要遍历两次链表。
如何在只遍历一次链表的情况下实现呢?
假设我们可以从后往前遍历链表,那么从尾节点向前走 k - 1 步就能走到倒数第 k 个节点,也就是说,倒数第 k 个节点和尾节点的距离是 k - 1。所以,我们可以维护两个指针,都从头节点开始,一个指针先不动,另一个指针往前走 k - 1 步,这样的话两个指针的距离就是 k - 1。之后两个指针一起往后走,当走在前面的指针到达尾节点时,走在后面的指针正好到达倒数第 k 个节点。
参考代码:
class Solution {
public ListNode getKthFromEnd(ListNode head, int k) {
// 使用间隔为 k 的双指针
if(head == null || k == 0)
return null;
ListNode pre = head;
ListNode post = head;
for(int i = 0; i < k - 1; i++){
if(pre.next != null)
pre = pre.next;
else
return null;
}
while(pre.next != null){
pre = pre.next;
post = post.next;
}
return post;
}
}