LeetCode 第 19 题 (Remove Nth Node From End of List)

LeetCode 第 19 题 (Remove Nth Node From End of List)

删除一个链表的倒数第 N 个元素。这个属于链表的常规操作。通常的做法是遍历两遍链表，第一遍时确定链表的元素总数，根据这个元素总数就可以算出倒数第 N 个元素的位置了。第二遍时删除这个元素。这道题不让遍历两遍，那么只能将链表各个元素的位置存下来了，遍历一遍之后，直接去删除那个元素。

删除链表元素时要分情况讨论。如果要求删除的这个元素根本就不在链表中，那么不用删除。如果这个元素是链表的第一个元素或最后一个元素也要特殊处理。

这些都考虑了就没有什么问题了。下面是代码：

ListNode* removeNthFromEnd(ListNode* head, int n)
{
if(n <= 0) return head;
vector<ListNode *> array;
ListNode * p = head;
while(p)
{
array.push_back(p);
p = p->next;
}
int m = array.size();
if(n > m) return head;
if(n == m) //如果是链表的第一个元素，要特殊处理
{
return head->next;
}
if(n == 1) //如果是链表的最后一个元素也要特殊处理
{
array[m - 2]->next = NULL;
return head;
}
array[m - n - 1]->next = array[m - n + 1];
return head;
}