LeetCode 19 - Remove Nth Node From End of List (Medium)

문제

LeetCode - 19번

풀이 과정

링크드 리스트에서 뒤에서 n 번째 노드를 삭제하는 연산을 구현하는 문제입니다.

뒤에서 n 번째 노드를 찾기 위해서 우선 리스트의 전체 크기를 알아낸 다음, 이동해야 하는 offset 을 계산합니다.
이후 두 개의 포인터를 활용해서 next 속성을 재지정하면 됩니다.

코드

/**
 * 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) {
  let offset = getSize(head) - n;
  let prev = null;
  let cur = head;

  while (offset > 0) {
    prev = cur;
    cur = cur.next;
    offset--;
  }

  if (prev) {
    prev.next = cur.next;
  } else {
    head = head.next;
  }

  return head;
};

function getSize(head) {
  let size = 1;

  for (let cur = head; cur.next; cur = cur.next) size++;

  return size;
}