[力扣]面试题22. 链表中倒数第k个节点

链表的倒数第k个节点

Posted by JinFei on April 6, 2020

题目描述

输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。

Example1:

  给定一个链表: 1->2->3->4->5, 和 k = 2.
    返回链表 4->5.

解题思路

  • 快,慢指针
/**
 * 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) {
        if(head == nullptr || k < 0){
            return nullptr;
        }
        if(k == 0){
            return head;
        }
        ListNode* fast = head;
        ListNode* slow = head;
        int step = k - 1;
        while(step > 0){
            fast = fast -> next;
            step--;
        }
        if(fast == nullptr){
            return nullptr;
        }
        while(fast && fast -> next){
            fast = fast -> next;
            slow = slow -> next;
        }
        return slow;
    }
};