代码改变世界

Remove Nth Node From End of List

2015-03-16 10:25  笨笨的老兔子  阅读(127)  评论(0编辑  收藏  举报

去掉一个单链表的倒数第N个节点

思路很简单,两个指针,一个指针提前跑n个节点,然后再一起奔跑,提前跑的那个到达尾部的时候,后面跑的那个指向倒数第n+1节点。
需要注意一些边界:

  • 头指针为空
  • n大于链表长度
  • n等于链表长度

    1. class Solution {
    2. public:
    3. ListNode *removeNthFromEnd(ListNode *head, int n) {
    4. if (!head )
    5. {
    6. return NULL;
    7. }
    8. ListNode* forward = head;
    9. ListNode* back = head;
    10. for (int i = 0; i < n ; i++)
    11. forward = forward->next;
    12. if (!forward)
    13. {
    14. return head->next;
    15. }
    16. while (forward->next != NULL)
    17. {
    18. forward = forward->next;
    19. back = back->next;
    20. }
    21. back->next = back->next->next;
    22. return head;
    23. }
    24. };