LC-19 删除链表倒数第N个节点

目标：

给出一个链表的head，和n，删除该链表倒数第n个节点，并返回head。尽量使用一个循环实现。

 

思路：

用一个指针index来遍历的链表，同时用一个整形count记录搜索节点的迭代次数，直到与题目输入的n相等。

这时候，用一个指针before_remove来记录应该被删的节点的前一个节点。

同时，这时候before_remove与index同时开始迭代。直到index为链表最后一个节点。

这时候，如果count与n相等，即证明before_remove记录的是正确的应该被删的节点的前一个节点。

如果count与n不相等，又因为题目说n的输入是绝对正确的，所以只能这时应该删除的节点是head，所以导致没有应该被删节点的前一个节点。

接下来就是节点的删除工作，分别对应2种情况。

 

代码：

/**
 * 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* before_remove = head;
        ListNode* index = head;
        int count = 0;
        /* find the node that before the node which should be removed */
        while (index -> next != NULL) {
            index = index -> next;
            if (count == n)
                before_remove = before_remove -> next;
            else
                count++;
        }
        /* make the pointer correct and remove the node */
        ListNode* should_remove;
        // normal situation
        if (count == n) {
            should_remove = before_remove -> next;
            before_remove -> next = should_remove -> next;
        }
        // the head node should be removed
        else {
            should_remove = head;
            head = head -> next;
        }
        delete should_remove;
        return head;
    }
};