删除链表中倒数第n个节点

描述

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

注意事项

链表中的节点个数大于等于n

样例

给出链表1->2->3->4->5->null和 n = 2.

删除倒数第二个节点之后，这个链表将变成1->2->3->5->null.

思考

1. 如果是第一个元素,直接删除返回 temp1
2. 如果不是，则找到前一个元素,删除后返回指向头结点的指针

代码

//  By Lentitude/** * Definition of ListNode * class ListNode { * public: *     int val; *     ListNode *next; *     ListNode(int val) { *         this->val = val; *         this->next = NULL; *     } * } */class Solution {public:    /**     * @param head: The first node of linked list.     * @param n: An integer.     * @return: The head of linked list.     */    ListNode *removeNthFromEnd(ListNode *head, int n) {        // write your code here        if (!head) return NULL;        ListNode *l1 = head, *temp1 = head;        int num = 0;    // 计算节点数量        while (l1){            ++num;            l1 = l1->next;        }        int index = 0;        while (temp1){            ++index;            /**             * 如果是第一个元素,直接删除返回 temp1             * 如果不是，则找到前一个元素,删除后返回指向头结点的指针             */            if (index == 1 && num == n){                temp1 = temp1->next;                return temp1;            }            if (index == num - n){                temp1->next = temp1->next->next;                return head;            }            temp1 = temp1->next;        }    }};