(链表)19. 删除链表的倒数第 N 个结点

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


示例 1:

(链表)19. 删除链表的倒数第 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]

提示:

  • 链表中结点的数目为 sz
  • 1 <= sz <= 30
  • 0 <= Node.val <= 100
  • 1 <= n <= sz

进阶:你能尝试使用一趟扫描实现吗?


代码:

/**
 * 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) {
        ListNode *dummyhead = new ListNode(-1);

        dummyhead->next = head;

        int cnt = -1;

        ListNode *cur = dummyhead;
        while(cur != nullptr){
            cnt++;
            cur = cur->next;
        }

        int num = cnt - n;

        cur = dummyhead;
        for(int i = num; i > 0; i--){
            cur = cur->next;
        }

        ListNode *tmp = cur->next;

        if(tmp == nullptr) return nullptr;

        cur->next = cur->next->next;

        head = dummyhead->next;

        delete tmp;
        delete dummyhead;
        
        return head;
    }
};

解题思路:

(1)首先,获取链表长度。

(2)接着,遍历到需要删除的前一个位置。

(3)最后,改变指针指向。

(4)注意,释放内存空间。

你可能感兴趣的:(力扣,链表,数据结构,算法)