给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
删除倒数第n个节点,用两个指针,先让后一个便利n个节点,然后两个指针一起往后走,当后头的走到最后,前头的下一个节点就是要删除的节点,然后把节点删掉,最后把原来链表head返回。
有一个特殊情况,就是要删除的节点是第一个节点,删掉之后head就空了。所以可以添加一个头节点,就不会出现这种特殊情况了, 结果返回头节点的next。
代码如下:
/**
* 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* headnode = new ListNode(0);
headnode->next = head;
ListNode* q = headnode;
ListNode* p = headnode;
for(int i = 0; i < n; i++)
q = q->next;
while(q->next != NULL)
{
q = q->next;
p = p->next;
}
q = p->next;
p->next = p->next->next;
delete(q);
ListNode* r = headnode->next;
delete(headnode);
return r;
}
};