给你一个链表,删除链表的倒数第 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
前言
在对链表进行操作时,一种常用的技巧是添加一个哑节点(dummy node),它的 next指针指向链表的头节点。这样一来,我们就不需要对头节点进行特殊的判断了。
例如,在本题中,如果我们要删除节点 y,我们需要知道节点 y 的前驱节点 x,并将 x的指针指向 y的后继节点。但由于头节点不存在前驱节点,因此我们需要在删除头节点时进行特殊判断。但如果我们添加了哑节点,那么头节点的前驱节点就是哑节点本身,此时我们就只需要考虑通用的情况即可。
特别地,在某些语言中,由于需要自行对内存进行管理。因此在实际的面试中,对于「是否需要释放被删除节点对应的空间」这一问题,我们需要和面试官进行积极的沟通以达成一致。下面的代码中默认不释放空间。
方法三:双指针
思路与算法
我们也可以在不预处理出链表的长度,以及使用常数空间的前提下解决本题。
由于我们需要找到倒数第 n个节点,因此我们可以使用两个指针 first和 second同时对链表进行遍历,并且 first比 second超前 n个节点。当 first遍历到链表的末尾时,second就恰好处于倒数第 n个节点。
具体地,初始时 first和 second均指向头节点。我们首先使用 first对链表进行遍历,遍历的次数为 n。此时,first 和 second 之间间隔了 n−1个节点,即 first比 second超前了 n个节点。
在这之后,我们同时使用 first 和 second对链表进行遍历。当 first遍历到链表的末尾(即 first为空指针)时,second恰好指向倒数第 n个节点。
根据方法一和方法二,如果我们能够得到的是倒数第 n个节点的前驱节点而不是倒数第 n 个节点的话,删除操作会更加方便。因此我们可以考虑在初始时将 second指向哑节点,其余的操作步骤不变。这样一来,当 first遍历到链表的末尾时,second的下一个节点就是我们需要删除的节点。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode *removeNthFromEnd(struct ListNode *head,int n)
{
struct ListNode *dummy = malloc(sizeof(struct ListNode));
dummy->val = 0,dummy->next = head;
struct ListNode *first = head;
struct ListNode *second = dummy;
for(int i=0;i<n;++i)
first = first->next;
while(first)
{
first = first->next;
second = second->next;
}
second->next = second->next->next;
struct ListNode *ans = dummy->next;
free(dummy);
return ans;
}