剑指Offer——JZ14 链表中倒数最后k个结点

原创于 2021-08-24 19:58:26 发布 · 176 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#链表

JZ 专栏收录该内容

17 篇文章

订阅专栏

这篇博客讨论了如何在C++中实现一个算法，从给定的链表中找到倒数第k个节点。提供的解决方案利用了栈的数据结构，先遍历整个链表并将所有节点压入栈中，然后弹出k个节点，最后栈顶的元素即为所求。如果链表长度小于k，则返回空链表。

描述

输入一个链表，输出一个链表，该输出链表包含原链表中从倒数第k个结点至尾节点的全部节点。
如果该链表长度小于k，请返回一个长度为 0 的链表。

输入：{1,2,3,4,5},1
返回值：{5}

/**
 * struct ListNode {
 *	int val;
 *	struct ListNode *next;
 *	ListNode(int x) : val(x), next(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param pHead ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    ListNode* FindKthToTail(ListNode* pHead, int k) {
        // write code here
        stack<ListNode*> stk;
        while(pHead){
        	stk.push(pHead);
        	pHead = pHead->next;
        }
        if(k > stk.size() || k == 0)
            return NULL;
        for(int i=0; i<k; i++){
        	stk.pop();
        }
        return stk.top();
    }
};