删除链表的倒数第n个节点(逆置链表)

• 描述

给定一个链表，删除链表的倒数第 n 个节点并返回链表的头指针
例如，

给出的链表为: 1→2→3→4→5,n=2.
删除了链表的倒数第 n 个节点之后,链表变为1→2→3→5.

 

数据范围： 链表长度 0≤n≤1000，链表中任意节点的值满足 0≤val≤100

要求：空间复杂度 O(1)，时间复杂度 O(n)
备注：

题目保证n 一定是有效的

 

• 示例

输入：{1,2},2

输出：{2} 

 

• 算法思想

先将链表逆置，采用头插法的思想逆置链表，然后删除正数第n个节点，最后再将链表逆置并返回

 

代码 

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* reverse(ListNode *head){
        ListNode *pre = new ListNode();
        ListNode *p = head;
        ListNode *r;
        while(p){
            r = p->next;
            p->next = pre->next;
            pre->next = p;
            p = r;
        }
        return pre->next;
    }
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        if(head->next==nullptr) return nullptr;
        ListNode *h = reverse(head);
        ListNode *p = h;
        ListNode *pre = new ListNode();//头结点
        pre->next = p;
        //删除第n个节点
        while(n>1){
            pre=p;
            p=p->next;
            n--;
        }
        if(pre->next == h){//如果删除的是第一个结点,要逆置的链表的第一个节点就不是h
            pre->next = p->next;
            delete p;
            return reverse(pre->next);
        }
        pre->next = p->next;
        delete p;
        return reverse(h);
    }
};