653. Two Sum IV - Input is a BST

2017-08-06  本文已影响23人  DrunkPian0

这周contest的签到题。我感觉应该有别的做法吧,但我直接把它遍历出来按照标准TWO SUM做的。。
而且在遍历的时候我竟然以为bst遍历要用preorder了。。。。MDZZ

    public boolean findTarget(TreeNode root, int k) {
        List<Integer> list = new ArrayList<>();
        preOrder(root, list);
        int i = 0, j = list.size() - 1;
        while (i < j) {
            if (list.get(i) + list.get(j) == k) {
                return true;
            }
            if (list.get(i) + list.get(j) < k) {
                i++;
            } else {
                j--;
            }
        }
        return false;
    }

    private void preOrder(TreeNode root, List<Integer> list) {
        if (root == null) {
            return;
        }
        preOrder(root.left, list);
        list.add(root.val);
        preOrder(root.right, list);
    }
上一篇 下一篇

猜你喜欢

热点阅读