Two Sum IV - 입력이 BST임

7893 단어 javascriptleetcode
이진 검색 트리의 루트와 대상 번호 k가 주어지면 BST에 두 개의 요소가 존재하여 합계가 주어진 대상과 같으면 true를 반환합니다.

예 1:
입력: root = [5,3,6,2,4,null,7], k = 9
출력: 참

예 2:
입력: root = [5,3,6,2,4,null,7], k = 28
출력: 거짓

접근법 1: 설명적이고 이해하기 쉬움

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @param {number} k
 * @return {boolean}
 */
var findTarget = function (root, k) {
  // collecting all Element form BST
  let allElements = [];
  // inorder traversal becoz it will give sorted array for simplicity
  function inOrderTraversal(root) {
    if (root === null) {
      return null;
    }
    inOrderTraversal(root.left);
    allElements.push(root.val);
    inOrderTraversal(root.right);
  }
  inOrderTraversal(root);
  // creating map to keep track of all elements
  let map = new Map();
  allElements.forEach((element, index) => {
    map.set(element, index);
  });
  // main loop to check if two values from array can give up the required sum
  for (let i = 0; i < allElements.length; i++) {
    let num = allElements[i];
    let diff = k - num;
    // checking if diff exist in map & also we have to make sure
    // index of current element is not same as of matching element
    if (map.has(diff) && map.get(diff) !== i) {
      return true;
    }
  }
  return false;
};



접근법 2: O(N)에서 수행하도록 위의 솔루션을 수정했습니다.

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @param {number} k
 * @return {boolean}
 */
var findTarget = function(root, k) {
     let map= new Map();
    let bool=false;

    const inorder =(root)=>{
        if(root===null) return;
        inorder(root.left);
        if(map.has(k-root.val)){
            bool=true;
        }else{
            map.set(root.val);
        }
        inorder(root.right);  
    }
   inorder(root);
   return bool;
};

좋은 웹페이지 즐겨찾기