(19)Remove Nth Node From End of List
题目:将一个链表的倒数第N个链节去掉。
例子:
所给的链表是: 1->2->3->4->5,同时 n = 2。
移除链节之后,链表为: 1->2->3->5。
根据题目,想到的就是通过从前向后计算链表长度然后去掉第len - n个链节就可以了。
我觉得可能会有更好的算法,就看了一下几个博客,使用的方法是将两个指针中间间隔n个,一起向后查找,直到后面的指针为NULL,但是我考虑之后觉得的确是一种好想法,但是却觉得复杂度差不多,所以就没改自己的代码,下面是我写的代码:
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
int len = 0;
if(n == 0){
return head;
}
ListNode *p = head, *q = head, *result = head;
while(p->next != NULL){
len ++;
p = p->next;
}
len ++;
if(n > len){
return head;
}
if(n == len){
return head->next;
}
for(int i = 1;i < len - n; i ++){
q = q->next;
}
q->next = q->next->next;
return result;
}
};