19.删除链表的倒数第 N 个结点

给你一个链表,删除链表的倒数第  n  个结点,并且返回链表的头结点。

示例 1:

输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]

示例 2:

输入:head = [1], n = 1
输出:[]

示例 3:

输入:head = [1,2], n = 1
输出:[1]

提示:

链表中结点的数目为 sz 1 <= sz <= 30 0 <= Node.val <= 100 1 <= n <= sz

**进阶:**你能尝试使用一趟扫描实现吗?

解法

/**
 * Definition for singly-linked list.
 * function ListNode(val, next) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.next = (next===undefined ? null : next)
 * }
 */
/**
 * @param {ListNode} head
 * @param {number} n
 * @return {ListNode}
 */
var removeNthFromEnd = function (head, n) {
  const getSize = (head) => {
    let size = 0;
    const helper = (node) => {
      if (node.next) {
        size++;
        return helper(node.next);
      } else {
        return size + 1;
      }
    };

    if (head === undefined) {
      return size;
    } else {
      return helper(head);
    }
  };

  let current = head;
  let index = 0;
  const length = getSize(head);
  const targetIndex = length - n;

  if (targetIndex === 0) {
    return head.next;
  }

  while (index < targetIndex - 1) {
    current = current.next;
    index++;
  }

  current.next = current.next.next;
  return head;
};