题目来自LeetCode,链接:面试题22. 链表中倒数第k个节点。具体描述为:输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。
示例:
给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.
这题目也是蛮典型的双指针的题目,先用一个后指针往后走k的长度,再让一个前指针指向头节点,那么现在让两个指针同时向后走,当后指针为空的时候,前指针指向的就是倒数第k个节点 。时间复杂度为 O ( n ) O(n) O(n),空间复杂度为 O ( 1 ) O(1) O(1)。
JAVA版代码如下:
/**
* 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) {
ListNode prev = head;
for (int i = 0; i < k; ++i) {
head = head.next;
}
while (head != null) {
head = head.next;
prev = prev.next;
}
return prev;
}
}
提交结果如下:
Python版代码如下:
# 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:
prev = head
for i in range(k):
head = head.next
while head is not None:
head = head.next
prev = prev.next
return prev
提交结果如下: