Given a linked list, remove the n th node from the end of list and return its head.
For example,
Given linked list: 1->2->3->4->5, and n = 2.
After removing the second node from the end, the linked list becomes 1->2->3->5.
Note:
Given n will always be valid.
Try to do this in one pass.
思路:
问题关键在于找到倒数第n个节点,并且尽量只使用一次循环。
采用两个指针,对前指针,使其先走出n步,随后两个指针同时前进,当前指针到达链表尾部时,后指针到达倒数第n个节点的位置。
注意:删除时注意待删除节点为头结点时的情况。
**特殊情况:
默认,当n大于链表的长度时,我默认删掉第一个元素。当然也可以直接返回原来的链表,只要自己考虑到这种特殊情况即可**
代码如下:
/**
* 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) {
ListNode *newHead=new ListNode(0);
newHead->next=head;
ListNode *slow=newHead,*fast=newHead;
//fast先移动n步
int i=0;
for(;i<n;i++)
{
if(fast->next!=NULL)//保证fast最远只能指向链表的最后一个元素。
fast=fast->next;
else//此时,表示n>=链表的长度,需要特殊处理,仅仅删除头结点即可
break;
}
if(i==n)//说明n小于等于链表的长度
{
//slow和fast同时移动,结束后slow指向待删除元素的前一个位置,fast指向最后一个元素
while(fast->next)
{
slow=slow->next;
fast=fast->next;
}
}
//删除待删除的元素
ListNode *tmp=slow->next;
slow->next=slow->next->next;
delete tmp;
return newHead->next;
}
};