删除链表中倒数第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;
        }

    }
};