牛客剑指Offer面试题22:(删除)链表中倒数第k个节点

1.优先考虑小的遍历次数,并注意代码的鲁棒性
(各种特殊条件:头指针为空;节点数为0;链表总节点数小于k等各种令代码崩溃的条件)
2.当一个指针遍历链表不能解决问题时,可考虑两个指针.一个每次一步,一个每次多步或一个提前先走若干步,在同时开始走

题目描述

输入一个链表,输出该链表中倒数第k个结点。

/*
struct ListNode {
	int val;
	struct ListNode *next;
	ListNode(int x) :
			val(x), next(NULL) {
	}
};*/
class Solution {
public:
    ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
        //判断有效性
        if(pListHead == nullptr || k == 0)
            return nullptr;
        //定义两个指针一前一后
        ListNode* pAhead = pListHead;
        ListNode* pBehind = nullptr;
        //
        for(int i = 0 ; i < k - 1 ; ++i)
        {
            //每走一步判断下一步是否为空,如节点数小于k则提前退出
            if(pAhead->next != nullptr)
                pAhead = pAhead->next;
            else
                return nullptr;
        }
        //前后指针相距k-1个节点,开始同时前进直至前指针到达尾结点
        pBehind = pListHead;
        while(pAhead->next != nullptr)
        {
            pAhead = pAhead->next;
            pBehind = pBehind->next;
        }
        return pBehind;
    }
};

若要求删除倒数第k个结点则需要预设一个哑结点(代码中pPre)(因为头结点可能被删除,因此预设哑结点并且最后返回哑结点的下一个结点),该节点的next指向head,且前后指针都从pPre出发,后指针走到欲删除结点的前一个结点

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        
        if(head->next == nullptr)
        {
            return nullptr;
        }
        //针对头结点可能被删的情况设定pPre用于最后的返回(pPre的next指向head)
        ListNode* pPre = new ListNode(0);
        pPre->next = head;
        ListNode* pFirst = pPre;
        ListNode* pSecond = pPre;
        //先走n步
        for(int i = 0; i < n ; ++i)
        {
            pFirst = pFirst->next;
        }
        
        //找到倒数n+1个结点
        while(pFirst->next != nullptr)
        {
            pFirst = pFirst->next;
            pSecond = pSecond->next;
        }
        pSecond->next = pSecond->next->next;
        return pPre->next;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值