给出链表1->2->3->4->5->null和 n = 2.
删除倒数第二个节点之后,这个链表将变成1->2->3->5->null.
这个题目相当于在“返回链表倒数第n个节点”的基础上增加功能,通过上一篇文章,我们已经有了寻找倒数第n个节点的方法。这里我们只提供遍历一次的方法,遍历2次也是这个道理。
删除节点k的关键在于要找到k的前一个节点,其次就是特殊情况的判断。代码如下:
<span style="color:#717171;">class Solution {
public:
/**
* @param head: The first node of linked list.
* @param n: An integer.
* @return: The head of linked list.
*/
ListNode *removeNthFromEnd(ListNode *head, int n) {
// write your code here
ListNode *p1, *p2, *pre;
if ( head == NULL || n <= 0 ) {
return NULL;
}
p1 = head;
p2 = head;
for ( int i = 0; i < n-1; i++ ) {
if ( p1 -> next != NULL ) {
p1 = p1 -> next;
} else {
return NULL;
}
}
while ( p1 -> next != NULL ) {
p1 = p1 -> next;
pre = p2; //关键一步:标记倒数第n个节点的前一个节点
p2 = p2 ->next;
}
if ( p2 == head ) { //关键判断:如果只有一个节点,倒数第n个就是第一个,删除后链表为空
head = head -> next;
} else {
pre ->next = pre -> next -> next;
}
return head;
}
};</span>