Leetcode 删除链表的倒数第N个节点

删除链表的倒数第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() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* nodes[100000]; // 用于存储节点地址
    int count; // 节点个数
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        countAndSave(head); // 初始化
        //删除节点
        if(n == count){ //特殊情况：删除第一个
            return head->next;
        }
        int pre = count - n - 1;
        ListNode* temp = nodes[pre+1];
        nodes[pre]->next = nodes[pre+2]; // 移除
        delete temp;
        return head;
    }
    void countAndSave(ListNode* head){
        this->count = 0;
        ListNode* temp = head;
        while(temp != nullptr){
            nodes[count++] = temp;
            temp = temp->next;
        }
    }
};

关键是要确定待删除的节点位置，因此这个节点个数就需要确定，而且如果遍历一次就要实现的话，就可以采用节点数组来存储每个节点的地址。