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

一、题目描述

给定一个链表，删除链表中倒数第n个节点，返回链表的头节点。
注意事项：链表中的节点个数大于等于n
样例
给出链表1->2->3->4->5->null和 n = 2.
删除倒数第二个节点之后，这个链表将变成1->2->3->5->null.
挑战
O(n)时间复杂度

题目链接：删除链表中倒数第n个结点

二、解题思路

如果先遍历链表，得出链表结点个数，然后再第二次遍历找出倒数第n个结点。这样做需要遍历两次链表，可不可以只遍历一次就找到倒数第n个结点呢？可以！
步骤：
1、我们定义两个指针，第一个指针ahead开始遍历到链表的第n-1个结点，第二个指针behind先不动；
2、从第k个结点开始，behind也开始从链表的头指针（第一个结点）开始遍历，直到ahead指针到达链表的尾部结点。这时，behind正好是倒数第n个结点。
3、删除这个结点。
当然还有特殊情况需要考虑：
1、删除的结点是头结点或者尾结点
2、链表只有1个结点
3、链表为空

三、代码实现

/**
 * Definition for ListNode.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int val) {
 *         this.val = val;
 *         this.next = null;
 *     }
 * }
 */ 
public class Solution {
    /**
     * @param head: The first node of linked list.
     * @param n: An integer.
     * @return: The head of linked list.
     */
    ListNode removeNthFromEnd(ListNode head, int n) {
        //特殊情况：头结点为空或链表只有一个结点或n<=0        
        if (head == null || head.next == null || n <= 0) {
            return null;
        }
        ListNode ahead = head;  //第一个指针
        ListNode behind = null; //第二个指针
        if (n == 1 && ahead.next != null) { //特殊情况：删除倒数第1个结点，即尾结点
            while (ahead.next.next != null) {   //找到倒数第二个结点
                ahead = ahead.next;
            }
            ahead.next = null;  //删除最后一个结点
            return head;    //返回头结点
        }
        for (int i = 0; i < n - 1 ; i++) {//将第一个指针后移n-1个位置
            if (ahead.next != null) {
                ahead = ahead.next;
            } else {    //链表结点数小于n的情况
                return null;
            }
        }
        behind = head;
        /**
        while循环结束后，因为两个指针中间相差n-1个位置，所以当ahead到达链表的尾结点之后，behind刚好到达倒数第n个结点
        */
        while (ahead.next != null) {//两个指针同时后移，直到ahead到达链表的尾结点
            ahead = ahead.next;
            behind = behind.next;
        }
        //将倒数第n个结点的下一个结点的值赋给倒数第n个结点，相当于删除倒数第n个结点
        behind.val = behind.next.val;
        //删除下一个结点
        behind.next = behind.next.next;
        //返回根结点
        return head;
    }
}