题目:
Given a link list,remove the n th node from the end of list and return its head.
Note:
Given n will always be valid.
Try to do this in one pass.
给一个单链表,让你删除它的倒数第n个结点。
这个题目很简单,可以先遍历一遍求出它的结点个数k,然后再遍历过去删除掉它的第k-n+1个结点就好,但是提示里面让你试着只遍历一遍来完成这个任务,就需要一点小小的创新了。
如图:
- 我们分别设置两个快慢指针,两个指针都从头节点开始,让快指针fast先走n步,如图上半部分,这样快指针fast就比慢指针slow领先了n个结点。
- 然后再让他们同时往链表尾部遍历,当快指针到达链表的最后一个节点时停下,这时,慢指针的下一个节点就是我们要删除的节点,我们直接让慢指针的next指针指向它的下下一个节点,就完成了删除。如图下半部分。
特殊情况处理
上面的方法可以应对一般的情况,但对于特殊情况还需要进行特殊处理。
- 当传过来的链表头指针head为空时,直接return NULL;因为没法继续玩了,直接返回。
- 当要删除的结点恰好是头结点时,我们只需要将head指针向后挪动一个节点就好了。如图
当快指针已经走到链表的最后一个节点了慢指针还在head节点没有动,那么要删的结点就是头结点了(因为题目保证了输入的n是合法的),让head = head->next;就完成了。
源代码
语言:Cpp
编译环境:VS2015
#include<iostream>
using namespace std;
struct ListNode { //结构体声明
int val; //数据
ListNode *next; //指向下一个节点的指针
ListNode(int x) : val(x), next(NULL) {} //构造函数
};
class Solution { //解决方案类
public:
ListNode *removeNthFromEnd(ListNode *head, int n) { //删除倒数第n个节点的函数
ListNode *fast = head,*slow = head;
if (head == NULL)return NULL;
while (fast = fast->next) //快指针走
{
if (n <= 0) //n等于0说明慢指针该走了
{
slow = slow->next;
}
n--;
}
if (n > 0)head = head->next; //快指针走到头了慢指针还没走,删掉头结点
else slow->next = slow->next->next; //一般情况,删掉慢指针的下一个节点
return head;
}
};
int main()
{
Solution S;
int n = 4; //删掉的是倒数第n个节点
ListNode N1(5); //节点定义
ListNode N2(4);
ListNode N3(3);
ListNode N4(2);
ListNode N5(1);
N1.next = &N2; //将节点连接为链表
N2.next = &N3;
N3.next = &N4;
N4.next = &N5;
S.removeNthFromEnd(&N1, n);
}
main函数随便写来测试的,测试用例可以随便更换。
编辑于2018-8-7 01:50:52