删除链表的倒数第N个节点(中等)
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 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) {
if (!head->next) return NULL;
ListNode *p=head;;
int num=1;
while(p->next!=NULL){
num++;
p=p->next;
}
p=head;
for(int i=1;i<num-n;i++){
p=p->next;
}
if(num==n)
head=head->next;
p->next=p->next->next;
return head;
}
};
如果要求一次遍历解决问题,那么就得想些比较巧妙的方法了。比如首先要考虑的时,如何找到倒数第N个节点,由于只允许一次遍历,所以不能用一次完整的遍历来统计链表中元素的个数,而是遍历到对应位置就应该移除了。那么就需要用两个指针来帮助解题,p 和 q指针。首先 p 指针先向前走N步,如果此时 p 指向空,说明N为链表的长度,则需要移除的为首元素,那么此时返回 head->next 即可,如果 p 存在,再继续往下走,此时 q 指针也跟着走,直到 p 为最后一个元素时停止,此时 q 指向要移除元素的前一个元素,再修改指针跳过需要移除的元素即可,参见代码如下:
/**
* 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) {
if (!head->next) return NULL;
ListNode *p,*q;
p=q=head;
for(int i=0;i<n;++i){
p=p->next;
}
if(p==NULL){
head=head->next;
return head;
}
while(p->next){
p=p->next;
q=q->next;
}
q->next=q->next->next;
return head;
}
};