IT博客汇
  • 首页
  • 精华
  • 技术
  • 设计
  • 资讯
  • 扯淡
  • 权利声明
  • 登录 注册

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

    summer发表于 2016-07-07 14:44:47
    love 0

    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;
    }



沪ICP备19023445号-2号
友情链接