19. Remove Nth Node From End of List
Given a linked list, remove the nth 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.
问题很简单,要从一个给定的List中删除倒数第n个元素。但是有一个要求:Try to do this in one pass.
ListNodo结构体如下:
// Definition for singly-linked list.
struct ListNode {
int val;
ListNode *next;
ListNode(int x) : val(x), next(NULL) {}
};
只能在O(n)的时间内,找到要删除的结点。可以使用两个ListNode变量实现。因为n
可以确定要删除的结点和该List中最后一个结点的距离,因此,我们设置两个ListNode变量p
和q
,它们的距离为n+1
。那么,当q
到达List尾部的空结点时,p
自然就到达了要删除结点的前一结点了。
此外,还需要考虑的一种异常情况是,n
等于List的长度时,要进行另外处理,否则p
与q
的距离有可能在遍历过程中发生变化,或出现非法访问。
代码实现如下:
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode *p = head, *q = head;
while (n-- >= 0) {
if (q == NULL) {
ListNode* temp = head;
head = head->next;
delete temp;
return head;
}
q = q->next;
}
while (q != NULL) {
q = q->next;
p = p->next;
}
ListNode* temp = p->next;
p->next = p->next->next;
delete temp;
return head;
}
};