leetcode19. 删除链表的倒数第 N 个结点

题目

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

示例 1：

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

示例 2：

输入：head = [1], n = 1
输出：[]

示例 3：

输入：head = [1,2], n = 1
输出：[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:
    int getLength(ListNode* head){
        int len = 0;
        while(head){
            len++;
            head = head->next;
        }
        return len;
    }

    ListNode* removeNthFromEnd(ListNode* head, int n) {
        int len = getLength(head);
        ListNode* dummy = new ListNode(0, head);
        ListNode* cur = dummy;
        // 0开始就报错，1开始就没事
        // 这里cur->next 刚好指向要删除的结点，不会为空，所以不用加判断条件
        for(int i = 1; i < len-n+1; i++){
            cur = cur->next;
        }
        cur->next = cur->next->next;

        ListNode* ans = dummy->next;
        delete dummy;
        return ans;
    }
};

learn

删除链表的操作