【力扣】19.删除链表的倒数第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]

提示：

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

AC代码

/**
 * 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 len = 0;
        for (ListNode* tem = head; tem; len++) {
            tem = tem->next;
        }
        //删除指定元素
        ListNode* tem1 = head;
        ListNode* tem2 = head;
        for (int i = 0; i < len; i++) {
            if (len == n) { //删除第一个节点
                head = head->next;
                return head;
            }
            if (i == len-n-1) { //删除别的节点
                tem2 = tem2->next;
                tem1->next = tem2->next;
                return head;
            }
            else { //向后遍历链表
                tem1 = tem1->next;
                tem2 = tem2->next;
            }
        }
        return head;
    }
};