LeetCode 19 Remove Nth Node From End of List

Remove Nth Node From End of List

O(n)复杂度

1.新建一个头节点作为虚拟头部，这样在删除过程中就不需要考虑一些特殊的情况(删除的就是第一个节点)
2.用两个指针p和q,当p已经走了n步的时候，在让pq都同时往后走，当p走到头为NULL的时候，q指向的就是倒数第n个数，过程中记录父亲节点，直接修改指针即可

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        ListNode *p, *q,*pre;
        ListNode * node = new ListNode(0); //增加一个虚拟的头节点
        node->next = head;
        pre = node;
        p = q = head;
        if(head==NULL) return NULL;
        int i=0;
        //这里加一个头节点方便理解，方便删除
        //用两个指针p和q,当p已经走了n步的时候，在让pq都同时往后走，当p走到头为NULL的时候，q指向的就是倒数第n个数
        //过程中记录父亲节点，直接修改指针即可
        while(p!=NULL){
            if(i>=n){
                pre = q;
                q= q->next;
            }
            i++;
            p=p->next;
        }
        pre->next = q->next;
        free(q);
        return node->next;
    }
};