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.

思路：

假设我们想要删除倒数第1个节点5，那么我们应该把指针停留在倒数第2个节点4处，因为我们要进行node.next = node.next.next的节点删除操作。由此可见，如果我们想要删除倒数第n个节点的话，我们应该把指针停留在倒数n+1个节点处。

我们需要两个指针，一个先走，一个后走。等前面走了n+1步之后，后面的指针再走，此时两个指针保持恒定的距离n+1一直行走，直到前面的指针达到链表尾部。而此时后面跟着的指针到链表尾部的距离为n+1，也就是处在倒数第n+1个节点处，它的next节点就是要删除的倒数第n个节点，接下来就可以进行节点删除操作。

注意：

• 一定要创建一个dummy节点让其next为head，因为有可能链表只有一个元素，删除之后就为空，处理起来就麻烦了。
• 当链表有环的时候，下面的程序不适用，因为会出现死循环。

public ListNode removeNthFromEnd(ListNode head, int n) {
    if(head==null) return null;
    ListNode dummy = new ListNode(0);
    dummy.next = head;
    ListNode h = dummy;
    while(head != null) {
        head = head.next;
        if(n < 0 || --n < 0) { // n<0的话就不用再自减了，防止整数溢出
            h = h.next;
        }
    }
    h.next = h.next.next;
    return dummy.next;
}

 

上面的代码还是不太好理解，下面的更好理解：

public ListNode removeNthFromEnd(ListNode head, int n) {
    ListNode dummy = new ListNode(0);
    dummy.next = head;
    ListNode node = dummy;
    while(n-->0) head = head.next;
    while(head != null) {
        head = head.next;
        node = node.next;
    }
    node.next = node.next.next;
    return dummy.next;
}

 

再补充个C++的代码：

ListNode* removeNthFromEnd(ListNode* head, int n) {
    ListNode *node=head, **cur = &head;
    while(n--) node = node->next;
    while(node) {
        node = node->next;
        cur = &(*cur)->next;
    }
    *cur = (*cur)->next;
    return head; 
}