牛客网-删除链表的倒数第n个节点

一、题目

题目描述
给定一个链表，删除链表的倒数第 nn 个节点并返回链表的头指针
例如，
给出的链表为: 1\to 2\to 3\to 4\to 51→2→3→4→5, n= 2n=2.
删除了链表的倒数第 nn 个节点之后,链表变为1\to 2\to 3\to 51→2→3→5.

备注：
题目保证 nn 一定是有效的
请给出请给出时间复杂度为\ O(n) O(n) 的算法
示例1
输入

{1,2},2    

返回值

{2} 

二、思路

1、先循环一遍找到总节点个数
2、再循环找到要删除那个节点的前一个节点，定义一个指针=删除的节点，要删除节点的前一个->next=删除节点->next，删除节点=nullptr；

三、代码

/**
 * struct ListNode {
 *	int val;
 *	struct ListNode *next;
 * };
 */

class Solution {
public:
    /**
     * 
     * @param head ListNode类 
     * @param n int整型 
     * @return ListNode类
     */
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        if(head==nullptr ||head->next==nullptr)
        {
            return nullptr;
        }
        int Count=0;
        ListNode *p=head;
        while(p)
        {
            Count++;
            p=p->next;
        }
        int pos=0;
        ListNode *node=head;
        if((Count-n)==0)
        {
            ListNode *phead=head->next;
            return phead;
        }
        else 
        {
            while(pos!=Count-n-1)
            {
                node=node->next;
                pos++;
            }
            ListNode *n=node->next;
            node->next=n->next;
            n=nullptr;
            return head;
        }  
    }
};