19. 删除链表的倒数第N个节点
题目描述
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
题解:
法一:
两次遍历。
第一次遍历统计节点数量 len
,第二次遍历到底 len-n
个节点时停止,然后删除节点即可。
注意删除头节点的情况。
法一代码:
/**
* Definition for singly-linked list.
* 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) {
int len = 0;
ListNode *p = head;
while ( p ) {
++len;
p = p->next;
}
if ( len == n ) return head->next;
int k = 0;
for ( p = head; p; p = p->next ) {
++k;
if ( k == len - n ) {
p->next = p->next->next;
break;
}
}
return head;
}
};
/*
时间:4ms,击败:93.00%
内存:10.3MB,击败:99.87%
*/
法二:
一次遍历。该方法流程如下:
first
指针从head
开始走n
步second
指针从head
开始和first
同步更新first
指针为空时停止
若是不带头节点的链表,则在 first
指针走 n
步后,若其为空,需要删除的是 head
节点,处理一下即可。
若是带头节点的链表,那么 head
可以当做普通节点操作,很方便。
不带头节点:
/**
* Definition for singly-linked list.
* 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 *first = head;
int k = 0;
while ( k < n ) {
first = first->next;
++k;
}
if ( !first ) return head->next;
ListNode *second = head, *pre = NULL;
while ( first ) {
first = first->next;
pre = second;
second = second->next;
}
pre->next = second->next;
return head;
}
};
/*
时间:0ms,击败:100.00%
内存:10.5MB,击败:97.29%
*/
带头节点:
/**
* Definition for singly-linked list.
* 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 *dummy = new ListNode();
dummy->next = head;
ListNode *first = dummy;
for ( int i = 0; i <= n; ++i ) first = first->next;
ListNode *second = dummy;
while ( first ) {
first = first->next;
second = second->next;
}
second->next = second->next->next;
return dummy->next;
}
};
/*
时间:0ms,击败:100.00%
内存:10.2MB,击败:99.99%
*/