22. 链表中倒数第k个节点

1. 描述

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

2. 例子

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

3. 题解

n 是 m_head 中的节点数
时间复杂度: O(n)
空间复杂度: O(1)

class Solution 
{
public:
    // ListNode* getKthFromEnd(ListNode* head, int k)
    ListNode* getKthFromEnd(ListNode *m_head, int k) 
    {
        int nodesCount = 0;
        auto faseIter = m_head;
        for(int i = 0; i < k && faseIter; i++, nodesCount++)
            faseIter = faseIter->next;

        if(!faseIter) return (nodesCount == k ? m_head : nullptr);

        auto slowIter = m_head;
        while(faseIter)
        {
            slowIter = slowIter->next;
            faseIter = faseIter->next;
        }

        return slowIter;
    }
};
comments powered by Disqus