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]

来源：力扣（LeetCode）
链接：leetcode-cn.com/problems/re…

分析

1. 使用快慢指针，快指针先走n个节点，然后快慢指针一起走，快指针指向空时，慢指针就指向倒数第n个节点，由于要删倒数第n个节点，因此需要快指针指向空时，慢指针指向倒数第n个节点的前驱节点。
2. 要删除的可能是首个节点(倒数第n个)，因此，需要加上一个Head头结点，方便删除(否则没有指针指向头)。

实现

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
 
struct ListNode* removeNthFromEnd(struct ListNode* head, int n)
{
    if (head == NULL) {
        return NULL;
    }
    struct ListNode *pHead = (struct ListNode*)malloc(sizeof(struct ListNode));
    pHead->next = head;
    pHead->val = 0;    // 随意赋值，用不着
    struct ListNode *p1 = head;
    struct ListNode *p2 = pHead;

    for (int i = 0; i < n; i++) {
        p1 = p1->next;
    }

    while (p1 != NULL) {
        p1 = p1->next;
        p2 = p2->next;
    }

    // 删除
    p2->next = p2->next->next;
    return pHead->next;
}