链表中倒数第k个节点

题目描述

来源于 https://leetcode-cn.com/

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

示例:

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

返回链表 4->5.

解法:

使用两个指针,第一个指针先走 k 步,然后第二个指针开始从链表头部向前走。这样两个指针之间差距就是 k。两个指针一齐向后走,当前一个到达链表尾部时,第二指针自然就指向倒数第 k 个节点。

class Solution {
public:
    ListNode* FindKthToTail(ListNode* head, unsigned int k) {
        ListNode *fast = head, *slow = head;

        for(int i=0; i < k; i++){
            if(!fast) return nullptr;
            fast = fast->next;
        }
        while(fast){
            fast = fast->next;
            slow = slow->next;
        }
        
        return slow;
    }
};