leetcode 19. 删除链表的倒数第N个节点(Remove Nth Node From End of List)

目录

• 题目描述：
• 示例：
• 进阶：
• 解法：

题目描述：

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

示例：

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

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

说明：
给定的 n 保证是有效的。

进阶：

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

---

解法：

/**
 * 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) {
        int sz = 0;
        ListNode* cur = head;
        while(cur){
            sz++;
            cur = cur->next;
        }
        n = sz - n;
        cur = head;
        if(n == 0){
            return head->next;
        }
        while(--n){
            cur = cur->next;
        }
        // cur is the previous node of the deleted node
        cur->next = cur->next->next;
        return head;
    }
};