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

提示

        LintCode中的相关算法题实现代码，可以在我的GitHub中下载。

题目需求

    

给定一个链表，删除链表中倒数第n个节点，返回链表的头节点。

样例

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

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

挑战

O(n)时间复杂度

解题思路

    使用双指针法。

实现代码

    

/**
 * Definition of singly-linked-list:
 * class ListNode {
 * public:
 *     int val;
 *     ListNode *next;
 *     ListNode(int val) {
 *        this->val = val;
 *        this->next = NULL;
 *     }
 * }
 */

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
        if(head==NULL) return NULL;
        int i=0;
        ListNode* p1=head;
        ListNode* p2=head;
        ListNode* prev=NULL;
        while(i<n)
        {
            p1=p1->next;
            i++;
        }
        while(p1)
        {
            p1=p1->next;
            prev=p2;
            p2=p2->next;
        }
        if(p2==head) return p2->next;
        prev->next=p2->next;
        return head;
    }
};