题目
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
思路
用双指针就可以一趟扫描实现链表节点的删除,但是需要注意对于第一个节点的删除,在实现过程中通过dummyHead来保证对于第一个节点的删除与对其它节点的删除操作相同,这样就保证了对第一个节点删除不会出错误。
代码
/**
* 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 *dummyHead = new ListNode(0);
dummyHead -> next = head;
ListNode *cur = head, *before = dummyHead, *last = head;
while(n > 0){
last = last -> next;
n--;
}
while(last != NULL){
before = before -> next;
cur = cur -> next;
last = last -> next;
}
before -> next = cur -> next;
delete cur;
return dummyHead -> next;
}
};