删除链表的倒数第N个结点

题目描述

来源：leetcode_hot100_19
给你一个链表，删除链表的倒数第 n 个结点，并且返回链表的头结点。

示例 1：

输入：head = [1,2,3,4,5], n = 2
输出：[1,2,3,5]

示例 2：
输入：head = [1], n = 1
输出：[]

示例 3：
输入：head = [1,2], n = 1
输出：[1]

提示：
链表中结点的数目为 sz
1 <= sz <= 30
0 <= Node.val <= 100
1 <= n <= sz

进阶：你能尝试使用一趟扫描实现吗？

正解一

思路：遍历一遍获得长度len，就是要删掉正数第len-n+1个结点，则从i=0开始计数刚好数到第len-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* removeNthFromEnd(ListNode* head, int n) {
        int len=0;
        ListNode* p=head;
        while(p!=nullptr){
            len++;
            p=p->next;
        }
        int rn=len-n;
        if(rn==0){//要排除只有一个结点的情况，否则直接还是链表本身，如[1],1
            return head->next;
        }
        p=head;
        ListNode* p1=new ListNode(0,p);//新建一个头结点，方便搞一个p的前序结点p1，这样到了该删除的结点同时就知道了他的前序结点是谁，好改变next
        for(int i=0;i<rn;i++){
            p1=p;
            p=p->next;
        }
        p1->next=p->next;
        return head;

    }
};

正解二

思路：利用栈的先进后出特性

class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        stack<ListNode*> stk;
        ListNode* p0=new ListNode(-1,head);//因为val>=0，所以-1当做标志，用于要求删除第一个结点时连上后续的开头，即将它放在第一个是永远不会被删的
        stk.push(p0);
        ListNode* p=head;
        while(p!=nullptr){
            stk.push(p);
            p=p->next;
        }
        int i=0;
        while(i<n-1){//弹出被删结点之前的所有结点
            stk.pop();
            i++;
        }
        p=stk.top();//p指针指向被删结点
        stk.pop();
        ListNode* p1=stk.top();//p1指向被删的前一个结点
        if(p1->val==-1){//如果删的是第一个结点
            return p->next;//直接返回第二个结点及以后
        }
        p1->next=p->next;//将被删的前一个结点连上被删的后一个结点
        return head;//返回开头
    }
};