【LeetCode - 536】从字符串生成二叉树

1、题目描述

2、解题思路

  本题使用递归求解。

  根据题目示例的提示可知，字符串第一个左括号之前的数字是根节点，接着两个连续的最大括号（如果有）分别为左子树和右子树，对左右子树进行同样的递归操作即可，具体看代码。

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 TreeNode str2tree(String s) {
        if (s.length() == 0) return null;
        StringBuilder value = new StringBuilder();
        TreeNode root = new TreeNode();
        int start = -1; // 第一个左括号的位置
        // 确定根节点的权值
        for (int i = 0; i < s.length(); i++) {
            if (s.charAt(i) == '(') {
                root.val = Integer.parseInt(value.toString());
                start = i;
                break;
            }
            if (i == s.length() - 1) {
                value.append(s.charAt(i));
                root.val = Integer.parseInt(value.toString());
                return root;
            }
            value.append(s.charAt(i));
        }
        // 对左右子树递归求解
        int count = 1;  // 遇到左括号则++，遇到右括号则--
        for (int i = start + 1; i < s.length(); i++) {
            if ('(' == s.charAt(i)) count++;
            if (')' == s.charAt(i)) count--;
            if (count == 0) {
                // 找到了左子树的区间
                root.left = str2tree(s.substring(start + 1, i));
                if (i != s.length() - 1) {
                    // 剩余的就是右子树的区间
                    root.right = str2tree(s.substring(i + 2, s.length() - 1));
                }
                break;  // 左右子树整合完毕，结束
            }
        }
        return root;
    }
}