给定一个链表,使用一趟扫描实现, 删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
解题思路:假设双指针 p
和 q
,当 q
指向末尾的 NULL
,p
与 q
之间相隔的元素个数为 n
时,那么删除掉 p
的下一个指针就完成了要求。
设置虚拟节点 dummyHead 指向 head
设定双指针 p 和 q,初始都指向虚拟节点 dummyHead
移动 q,直到 p 与 q 之间相隔的元素个数为 n
同时移动 p 与 q,直到 q 指向的为 NULL
将 p 的下一个节点指向下下个节点
***最后不能直接返回head节点,而要新建一个retNode返回是为了防止删除头节点,而头结点没有前一个节点,所以设置了一个虚拟节点。
***
1、删除一个节点的关键是找到该节点的前向节点
2、头节点没有前向节点,就人为的添加一个前向的节点,这样可以使处理所有的情况实现统一,而不用因处理特殊情况引入复杂的开销
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* dummyHead = new ListNode(0);
dummyHead->next = head;
ListNode* p = dummyHead;
ListNode* q = dummyHead;
for( int i = 0 ; i < n + 1 ; i ++ ){
q = q->next;
}
while(q){
p = p->next;
q = q->next;
}
ListNode* delNode = p->next;
p->next = delNode->next;
delete delNode;
ListNode* retNode = dummyHead->next;
delete dummyHead;
return retNode;
}
};