LeetCode刷题笔记：653.两数之和 IV-输入BST

1. 问题描述

给定一个二叉搜索树 root 和一个目标结果 k，如果 BST 中存在两个元素且它们的和等于给定的目标结果，则返回 true。

2. 解题思路

第一步：层序遍历取出所有树节点的值，存储在一个 ArrayList 中
第二步：将 ArrayList 转为一个普通数组
第三步：像普通的两数之和一样处理

3. 代码实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public boolean findTarget(TreeNode root, int k) {
        if (root == null) {
            return false;
        }

        // 双端队列，存放树节点，层序遍历
        Queue<TreeNode> q = new LinkedList<>();
        q.offer(root);

        // 存放树节点的元素值
        List<Integer> list = new ArrayList<>();

        while (! q.isEmpty()) {
            // 记录每一层的节点数
            int n = q.size();
            // 遍历当前层
            for (int i = 0; i < n; i++) {
                TreeNode temp_node = q.poll();
                list.add(temp_node.val);
                if (temp_node.left != null) {
                    q.offer(temp_node.left);
                }
                if (temp_node.right != null) {
                    q.offer(temp_node.right);
                }
            }
            
        }

        int[] nums = new int[list.size()];
        for (int i = 0; i < nums.length; i++) {
            nums[i] = list.get(i);
        }

        for (int i = 0; i < nums.length - 1; i++) {
            for (int j = i + 1; j < nums.length; j++) {
                if (nums[i] + nums[j] == k) {
                    return true;
                } 
            }
        }
        return false;
    }
}