LeetCode 19.删除链表的倒数第N个结点 C++

题目描述

给你一个链表，删除链表的倒数第 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list

解题思路

先定义一个带有投结点的指针，头结点之后指向head。
然后通过栈的方式来实现寻找到倒数第N个结点。
1.将所有的结点全部入栈。
2.然后弹栈，直至弹到栈顶元素为要删除的结点的前一个结点。
3.将此时的栈顶元素即要删除的结点的前一个结点，所以我们只要将栈顶元素的next指针指向要删除的结点元素的后一个结点。即te->next=te->next->next，此时te为栈顶元素。这样就完成了删除结点的操作。
4.由于我们之前定义的链表是带头结点的，所以返回re->next即可。

代码

/**
 * 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) {
        ListNode* re=new ListNode(0,head);//带头结点的链表，并且头结点后面指向head
        ListNode* p=re;//移动指针
        stack<ListNode*> stk;//栈
        while(p!=NULL){//遍历链表
            stk.push(p);//入栈
            p=p->next;//指针后移
        }
        while(n>0){//从链表尾部开始弹栈，直至弹出倒数第n个结点。
            stk.pop();
            n--;
        }
        ListNode* te=stk.top();//记录此时的栈顶元素，即体重倒数第n个结点的前驱结点
        te->next=te->next->next;//将其与倒数第n个结点的后继结点相连，即完成删除操作
        return re->next;//返回re->next，因为re是带有头结点的
    }
};