剑指OFFER----面试题22. 链表中倒数第k个节点

Posted 2021-03-08 clown9804

链接：https://leetcode-cn.com/problems/lian-biao-zhong-dao-shu-di-kge-jie-dian-lcof/

 

代码：

/**
 * 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) {
        int n = 0;
        for (ListNode* p = head; p; p = p->next) n++;
        if (k > n) return nullptr;
        auto p = head; 
        for (int i = 0; i < n - k; ++i) p = p->next;
        return p;
    }
};