删除链表的倒数第N个节点
题目
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
思路
删除倒数第n个结点相当于删除正数第L-n-1个数,其中L是链表的长度。那么遍历链表拿到链表的长度,跳过第L-n-1个数直接指向下一个元素即可。时间复杂度为O(L)。
代码
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n)
{
ListNode *dummy=new ListNode(0);
dummy->next=head;
ListNode *first=head;
int L=0;
while(first)
{
L++;
first=first->next;
}
L=L-n;
first=dummy;
while(L)
{
L--;
first=first->next;
}
first->next=first->next->next;
return dummy->next;
}
};
其中的dummy指针用来记录first的位置,避免遍历后找不到表头的情况。
进阶
你能尝试使用一趟扫描实现吗?
上述思路是扫描了一遍链表得到长度,再遍历了一遍,共两遍。进阶要求只使用一趟扫描实现。
思路
这里定义两个指针, 第一个指针first先向前走n个位置,让second指针和first指针保持n的距离,然后一起移动两个指针,当first走到末尾的时候,second也就走到了倒数第n个的位置,再跳过即可。
代码
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n)
{
ListNode *dummy=new ListNode(0);
dummy->next=head;
ListNode *first=dummy;
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;
return dummy->next;
}
};
奇怪的知识又增长了,双指针真的有点强。