题目描述
输入一个链表,输出该链表中倒数第 k 个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。
LeetCode:链表中倒数第k个节点
示例:
给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.
解题思路
题目要求找出链表中倒数第 k 个节点,也即第 n - k + 1 个节点因为是单项链表,所以可以先从头开始遍历一遍得到链表的长度 n,再遍历一遍找到第 n - k + 1 个节点。该方法需要遍历两边链表。
我们也可以只遍历一遍得到想要的结果,方法就是设置两个指针,先让第一个指针走 k - 1 步,第二个指针保持不动,从第 k 步开始,第二个指针也开始遍历。此外,要考虑特殊值,如若 k > n 怎么办;k = 0 时呢; 头指针为 null 时呢?
由于两个指针的距离保持在 k - 1,当走在前面的指针到达链表的尾节点时,第二个指针正好指向想要的倒数第 k 个节点。
代码
public ListNode getKthFromEnd(ListNode head, int k) {
if (head == null || k == 0)
return null;
ListNode pFirst = head;
ListNode pLast = null;
for (int i = 0; i < k - 1; i++) {
// k > n 的情况
if (pFirst.next != null) {
pFirst = pFirst.next;
} else
return null;
}
pLast = head;
while (pFirst.next != null) {
pFirst = pFirst.next;
pLast = pLast.next;
}
return pLast;
}