给定一个链表,删除链表的倒数第 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]
代码如下:
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode* prehead=new ListNode(0, head);//在头节点之前定义一个哑节点
ListNode* prev=prehead;
stack<ListNode*> stk;//定义一个栈
while(prev)
{
stk.push(prev);//将链表中的元素入栈
prev=prev->next;//遍历链表
}
for(int i=0;i<n;i++)
{
stk.pop();//将栈中元素出栈,出栈完成后,此时栈顶元素就是要删除节点的前驱节点
}
ListNode* cur=stk.top();//定义删除节点的前驱节点
cur->next=cur->next->next;//删除倒数第n个节点
ListNode* ans=prehead->next;//保存哑节点的下一个结点,最终要返回它
delete prehead;//删除哑节点,哑结点只是为了方便头节点的使用,害怕头节点的丢失
return ans;//返回哑结点的下一个结点,链表的头节点
}
};