【LeetCode】19. 删除链表的倒数第N个节点

题目描述

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

示例：

给定一个链表: 1->2->3->4->5, 和 n = 2.

当删除了倒数第二个节点后，链表变为 1->2->3->5.

说明：

给定的 n 保证是有效的。

进阶：

你能尝试使用一趟扫描实现吗？

 

思路

为了一趟扫描实现，first 从头开始遍历到 n，若 first->next 为空，证明要删除的为第一个元素，返回 head->next 即可；

此时再定义一个指向头结点的指针 second，两个指针同时往后走，当 first 到末尾时，second 停位置的下一个即为想要删除的结点，second->next = second->next->next 即可。

 

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* removeNthFromEnd(ListNode* head, int n) {
        ListNode* first = head;
        for(int i=0; i<n; i++){
            first = first->next;
        }
        if(!first)
            return head->next;
        ListNode* second = head;
        while(first->next){
            second = second->next;
            first = first->next;
        }
        second->next = second->next->next;
        return head;
    }
};