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

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

给你一个链表，删除链表的倒数第 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* removeNthFromEnd(ListNode* head, int n) {
        int length=0;
        ListNode* tmp=head;
        while(tmp){
            length++;
            tmp=tmp->next;
        }
        ListNode* key=head;
        int count=length-n-1;
        if(count==-1){
            head=head->next;
            return head;
        }
        for(int i=0;i<count;i++){
            key=key->next;
        }
        ListNode* q=key->next;
        if(q && q->next){
            key->next=q->next;
        }else{
            key->next=nullptr;
        }
        delete q;
        return head;
    }
};