LintCode 删除链表中倒数第n个节点

来源：互联网发布：淘宝手机端详情页gif 编辑：程序博客网时间：2024/06/05 01:17

给出链表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>

0 0