题目背景:
难度:中等
要求:删除给定链表的倒数第N个节点,返回链表的头节点。
例如:
给定 head=[1,2,3,4,5] n=2
输出 [1,2,3,5]
**---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
算法思想:
伪头+快慢指针
(设置伪头节点的技巧:在初始化真正的头节点时不知道头节点值,可以设置一个伪头节点用来指向真正的头节点
,从而可以建立后面的建立链表过程,但是需要返回dummyHead->next
😉
1.设置一个伪头节点
2.设置快慢指针,让快指针先走,走到比慢指针快n+1步;
3.此时,让快慢指针一起前进,直到快指针指向尾节点的下一个,即快指针指向空的时候停止前进;
4.停止的时候,慢指针指向的下一个节点就是我们要删除的节点。删除该节点的方式是用其下一个节点对其进行覆盖;
代码实现:
//单链表定义1 自己要记住单链表的定义 面试的时候会自己写才行
struct ListNode
{
int val;//节点上存储的元素
ListNode* next;//指向下一个节点的指针
ListNode(int x):val(x),next(NULL){}//节点构造函数
};
//单链表定义的第二种方式
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* dummyHead = new ListNode(0);//伪头节点
dummyHead->next = head;
ListNode* slow = dummyHead;//快慢指针
ListNode* fast = dummyHead;
for (int i = 0; i < n; i++)
{
fast = fast->next;
}
fast = fast->next;//fast比slow多走n+1个节点
while (fast != NULL)
{
fast = fast->next;//快慢指针一起前进
slow = slow->next;
}
slow->next = slow->next->next;//要删除的节点是slow->next 删除的方法是用slow->next->next对slow->next进行覆盖;
return dummyHead->next;//返回真正的头节点
}
};
在这里插入代码片
感受:
1.要学会自己写单链表的定义方式
2.学会对伪头节点的使用,很多地方都会用到这个点
3.快慢指针步长不一样可以实现不同的效果
因为最近在学Go语言,找来了这道题的Go版本,感受一下和C++的区别:
func removeNthFromEnd(head *ListNode, n int) *ListNode {
dummy := &ListNode{0, head}
first, second := head, dummy
for i := 0; i < n; i++ {
first = first.Next
}
for ; first != nil; first = first.Next {
second = second.Next
}
second.Next = second.Next.Next
return dummy.Next
}