1.移除单链表中指定位置的节点

题目描述

给定一个链表，删除链表的倒数第 nn 个节点，并且返回链表的头结点。

原题链接

https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list/description/?utm_source=LCUS&utm_medium=ip_redirect&utm_campaign=transfer2china

示例1

输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
 
输入:head[1], n=1
输出:[]
 
输入:head = [1,2], n = 1
输出:[1]

进阶要求

尝试用一趟扫描实现

算法1

两次遍历

1. 第一次遍历求出链表长度
2. 第二次遍历删掉指定的结点
3. 注意删掉头结点的特殊情况(这个自己没有考虑到，当提交调错的时候才解决掉这个问题)

时间复杂度

遍历两次链表，空间复杂度为O(L)

空间复杂度

仅需要定义常数个指针变量，故时间复杂度为O(1)

代码

/**
 * 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 sum = 0;
        ListNode* a = head;
 
        // 求一下链表的长度
        while (a) {
            ++sum;
            a = a->next;
        }
        
        // 判断删掉第一个节点的特殊情况
        if (sum == n) {
            head = head->next;
            return head;
        }
 
        ListNode* b = head;
 
        // 遍历删掉指定的节点
        for (int i = 0; i < sum - n - 1; i++) {
            head = head->next;
        }
        head->next = head->next->next;
 
        return b;
    }
};

算法2

一次遍历

1.在头部之前添加保护点
2.设置两个指针first和second，均指向保护节点
3.first指针先向后移动n + 1个结点
4.然后first和second指针同时向后移动，直到first指针指向空，此时second结点指向的下一个结点需要删除

原理

始终保持两个指针之间间隔n个结点，在first到达终点时，second的下一个结点就是从结尾数第n个结点。

时间复杂度

遍历两次链表，空间复杂度为O(L)

空间复杂度

仅需要定义常数个指针变量，故时间复杂度为O(1)

class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        ListNode* e_head = new ListNode(0);
        e_head->next = head;                                            
 
        ListNode* first = e_head;
        ListNode* second = e_head;
 
        n++;
        while (n--) {
            first = first->next;
        }
 
        while (first) {
            first = first->next;
            second = second->next;
        }
        second->next = second->next->next;
 
        return e_head->next;
    }
};