删除链表的倒数第N个节点
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
尝试使用一趟扫描实现
使用双指针a,b,a首先向后移动n-1次,此时b指针不动,当a指针移动n-1次后,b开始向后移动,当a遍历完毕时,b恰好指向要删除的节点
代码实现:
/**
* 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) {
//双指针,a指针先走n-1步,b不动。b指针在a指针走n-1步之后从头开始走
//当a指针遍历完时,b指针指向倒数第n个节点
if(head->next==NULL)return NULL;
ListNode *a=head;
ListNode *b=head;
ListNode *te=head;
int count=0;
while(a->next!=NULL)
{
count++;
if(count>n-1)
b=b->next;
if(count>n)
te=te->next;
a=a->next;
}
if(b==head)head=head->next;
else te->next=b->next;
return head;
}
};