题目
给你一个链表,删除链表的倒数第 n 个结点,并且返回链表的头结点。
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
思路
- 遍历单指针,重点是理解虚拟节点dummy;在链表的头部放入一个哨兵,然后连上head节点,把head节点当做普通节点放心使用;
ListNode* dummy=new ListNode(0, head);
让它的指针指向链表的头节点,这样在删除节点的时候不需要判断是头结点;
这里new dummy的含义:dummy->val = 0, dummy->next = head;
即:数据域为0,指针域指向链表的头结点
第一次遍历统计链表中的元素个数 i;
遍历到第 i-n+1 个节点时,它的下一个节点就是我们需要删除的节点;
将下一个节点与第n个结点的后继结点相连;删除哨兵即空的头节点,返回链表。
- 快慢指针,快指针比慢指针先前进n个节点;当fast遍历到链表的末尾(即fast为空指针)时,slow恰好指向倒数第 n个节点。
在对链表进行操作时,一种常用的技巧是添加一个哑节点(dummy node),它的 \textit{next}next 指针指向链表的头节点。这样一来,我们就不需要对头节点进行特殊的判断了。
例如,在本题中,如果我们要删除节点 yy,我们需要知道节点 yy 的前驱节点 xx,并将 xx 的指针指向 yy 的后继节点。但由于头节点不存在前驱节点,因此我们需要在删除头节点时进行特殊判断。但如果我们添加了哑节点,那么头节点的前驱节点就是哑节点本身,此时我们就只需要考虑通用的情况即可。
特别地,在某些语言中,由于需要自行对内存进行管理。因此在实际的面试中,对于「是否需要释放被删除节点对应的空间」这一问题,我们需要和面试官进行积极的沟通以达成一致。
来源:力扣(LeetCode)
代码
方法一: 遍历单指针
class Solution { public: ListNode* removeNthFromEnd(ListNode* head, int n) { ListNode* dummy=new ListNode(0, head); int i = 0; ListNode* countnode = head; for(;countnode!=nullptr && countnode->next!=nullptr;i++){ countnode = countnode->next; } ListNode* cur = dummy; for(int j = 0;j < (i-n+1);j++){ cur = cur->next; } cur->next = cur->next->next; ListNode* ans = dummy->next; delete dummy; return ans; } };
方法二:快慢指针
class Solution { public: ListNode* removeNthFromEnd(ListNode* head, int n) { ListNode* dummy = new ListNode(0, head); ListNode* fast = head; ListNode* slow = dummy; for(int i = 0; i < n; i++){ fast = fast->next; } while(fast){ slow = slow->next; fast = fast->next; } slow->next = slow->next->next; ListNode* ans = dummy->next; delete dummy; return ans; } };