剑指 Offer 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个位置以后双指针一起先后运行当快指针到达链表尾部时,则慢指针的位置即为倒数第K个位置;
C++解法:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* getKthFromEnd(ListNode* head, int k) {
ListNode *p = head; //快指针
while(k){
p = p->next;
k--;
}
while(p != NULL){
p = p->next;
head = head->next;
}
return head;
}
};
python3解法
# 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:
slow_point = head
fast_point = head
for i in range(k):
fast_point = fast_point.next
while fast_point != None:
fast_point = fast_point.next
slow_point = slow_point.next
return slow_point