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;
}
};