LintCode 删除链表中倒数第n个节点

给出链表1->2->3->4->5->null和 n = 2.

删除倒数第二个节点之后，这个链表将变成1->2->3->5->null.

这个题目相当于在“返回链表倒数第n个节点”的基础上增加功能，通过上一篇文章，我们已经有了寻找倒数第n个节点的方法。这里我们只提供遍历一次的方法，遍历2次也是这个道理。

删除节点k的关键在于要找到k的前一个节点，其次就是特殊情况的判断。代码如下：

class Solution {
public:
    /**
     * @param head: The first node of linked list.
     * @param n: An integer.
     * @return: The head of linked list.
     */
    ListNode *removeNthFromEnd(ListNode *head, int n) {
        // write your code here
        ListNode *p1, *p2, *pre;
        if ( head == NULL || n <= 0 ) {
            return NULL;
        }
        p1 = head;
        p2 = head;
        for ( int i = 0; i < n-1; i++ ) {
            if ( p1 -> next != NULL ) {
                p1 = p1 -> next;
            } else {
                return NULL;
            }
        }
        while ( p1 -> next != NULL ) {
            p1 = p1 -> next;
            pre = p2;        //关键一步：标记倒数第n个节点的前一个节点
            p2 = p2 ->next;
        }
        if ( p2 == head ) {  //关键判断：如果只有一个节点，倒数第n个就是第一个，删除后链表为空
            head = head -> next;
        } else {
        pre ->next = pre -> next -> next;
        }
        return head;
    }
};