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

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

删除一个链表的倒数第 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;
}