LeetCode|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.

【解答】

计算链表长度，要删除元素的位置在length-n处。

两遍遍历：第一遍求链表长度，第二遍进行删除操作。

struct ListNode {
int val;
ListNode *next;
ListNode(int x) : val(x), next(NULL) {}
};
ListNode* insert(ListNode* ln,int val)
{
return (ln->next=new ListNode(val));
}

void print(ListNode* ln)
{
while(ln!=NULL)
{
cout<<ln->val<<" ";
ln=ln->next;
}
}
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
int length=0;
ListNode* ln=head;

if(head==NULL || head->next==NULL)
return NULL;
while(ln!=NULL)
{
length++;
ln=ln->next;
}
int NthFromHead=length-n;
int index=0;
ln=head;
if(index==NthFromHead)
{
head=head->next;
}
else{
while(index!=NthFromHead-1)
{
ln=ln->next;
index++;

}
ln->next=ln->next->next;
}

return head;

}
};

int main() {
Solution s;
ListNode* ln=new ListNode(1);
ListNode* head=ln;
for(int i=2;i<3;i++)
{
ln=insert(ln,i);
}
int n=2;
ListNode* rln=s.removeNthFromEnd(head,n);
print(rln);
return 0;
}

【遇到问题】

错误1：输入 [1] 1，对于只含有一个元素的单链表不能正确处理，由于访问了其next元素，导致程序崩溃。添加判断语句，如果输入为空或者链表中只有一个元素则直接返回NULL。

错误2：输入[1,2] 2，

index!=NthFromHead-1

index表示要删除元素的上一个元素。对于单链表来说，如果要删除元素为第一个元素，此时index==NthFromHead。这时候我们不应该向后寻找，而殷应该直接删除第一个元素。添加添加判断解决，是的head直接指向其next元素。

【改进程序】

采用双指针，两个指针位置相距n-1；当其中一个到达单链表尾部时，另一个指针正好在要删除元素的前一个位置。

class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* ln1=head;
ListNode* ln2=head;
if(head==NULL || head->next==NULL)
return NULL;
for(int i=0;i<n;i++)
ln1=ln1->next;
if(ln1==NULL)//确定要删除的是首部元素
head=head->next;
else
{
while(ln1->next!=NULL)
{
ln1=ln1->next;
ln2=ln2->next;
}
ln2->next=ln2->next->next;
}
return head;

}
};

【遇到问题】

while(ln1->next!=NULL)最初错写为while(ln1!=NULL)，导致输入为[1,2] 1时不能得到正确答案。

while循环应保证ln2正好指向要删除元素的前一个位置。