链表中倒数第k个节点

输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。

例如,一个链表有 6 个节点,从头节点开始,它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。

示例:

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

思路:

两个指针,一前一后,前面的指针先走k步,然后后面的指针跟进一起走,当前面的指针为nullptr时,后面的指针就是答案。

代码:

/**
 * 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 < 1) return nullptr;
        ListNode* dummyNode = new ListNode();
        dummyNode->next = head;
        ListNode* pre = dummyNode;
        ListNode* cur = head;
        for(int i = 1;i < k;i++) {
            if(cur->next == nullptr) {
                return nullptr;
            }
            else {
                cur = cur->next;
            }
        }
        while(cur) {
            cur = cur->next;
            pre = pre->next;
        }
        return pre;
    }
};