LeetCode 904 - Fruit Into Baskets (Medium)

문제

LeetCode - 904번

풀이 과정

최대 2종류의 과일을 담을 수 있는 바구니를 이용해서 가능한 많은 과일을 담은 경우를 찾는 투 포인터 문제입니다.

head 와 tail 을 이용해서 과일을 하나씩 순회하는데, 2종류의 제약조건을 만족시키는 과일이면 tail 을 증가시키고
그렇지 않을 경우 이를 만족할때까지 head 를 증가시키는 방법을 이용해서 구현합니다.

코드

/**
 * @param {number[]} tree
 * @return {number}
 */
var totalFruit = function (tree) {
  let head = 0;
  let tail = 0;
  let answer = 1;
  const basket = new Map();

  basket.set(tree[head], 1);

  while (head <= tail) {
    tail += 1;
    if (tail >= tree.length) {
      break;
    }

    const fruit = tree[tail];
    basket.set(fruit, basket.get(fruit) + 1 || 1);

    // if overload
    while (basket.size > 2) {
      const fruit = tree[head];

      basket.set(fruit, basket.get(fruit) - 1);
      if (basket.get(fruit) === 0) {
        basket.delete(fruit);
      }

      head += 1;
    }

    answer = Math.max(answer, tail - head + 1);
  }

  return answer;
};