输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。
示例:
给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/lian-biao-zhong-dao-shu-di-kge-jie-dian-lcof
解题思路
使用两个指针, 让其中一个指针先走k - 1步(走就是往后迭代)
此后两个指针同时走, 当后面的指针走到尾节点时, 前面的指针刚好是倒数第k个节点
注意:需要先算出长度, 因为如果k大于长度, 是不合理的
代码
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode getKthFromEnd(ListNode head, int k) {
if (head == null) {
return null;
}
int length = 1;
ListNode cur = head;
// 算出长度
while (cur.next != null) {
length++;
cur = cur.next;
}
// k大于长度, 不合理
if (k > length) {
return null;
}
// 定义两个指针
cur = head;
ListNode behind = head;
// 让其中一个指针先走k - 1步
for (int i = 0; i < k - 1; i++) {
behind = behind.next;
}
// 此后两个指针同时走, 当后面的指针走到尾节点时, 前面的指针刚好是倒数第k个节点
while (behind.next != null) {
behind = behind.next;
cur = cur.next;
}
return cur;
}
}