LeetCode - 删除链表的倒数第N个节点

题目

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

例如，

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

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

说明:

给的 n 始终是有效的。

尝试一次遍历实现。

解法

很简单的一道题，既然只允许遍历一次，且N一直有效，那么可以利用双指针解法（前指针、后指针），让前指针先走N步，再让两个在指针同时后移，直到前指针到达尾部，此时，后指针的下一个节点就是要被删除的节点了。

@Test
public void test() {
    ListNode head = new ListNode(1);
    ListNode listNode = removeNthFromEnd(head, 1);
    System.out.println(listNode);

}

public ListNode removeNthFromEnd(ListNode head, int n) {

    ListNode preNode = head;
    ListNode curNode = head;

    for (int i = 0; i < n; i++) {
        curNode = curNode.next;
    }

    if (curNode == null) {
        return preNode.next;
    }

    while (curNode.next != null) {
        preNode = preNode.next;
        curNode = curNode.next;
    }

    preNode.next = preNode.next.next;

    return head;
}

class ListNode {
    int val;
    ListNode next;

    ListNode(int x) {
        val = x;
    }
}