LeetCode 606 根据二叉树创建字符串[二叉树 dfs] HERODING的LeetCode之路

Posted 2022-03-21 HERODING23

解题思路：
一道变型的二叉树前序遍历，加上了模拟的思想，根据条件进行判断，分为左右两节点都在、左节点在和右节点在三个条件，根据情况插入双括号，代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode 
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) 
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) 
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) 
 * ;
 */
class Solution 
private:
    string ans;
public:
    string tree2str(TreeNode* root) 
        dfs(root);
        return ans;
    

    void dfs(TreeNode* node) 
        if(node == nullptr) return;
        ans += to_string(node->val);
        // 左右子树都在
        if(node->left != nullptr && node->right != nullptr) 
            ans += '(';
            dfs(node->left);
            ans += ")(";
            dfs(node->right);
            ans += ')';
        
        // 只有左子树在
        if(node->left != nullptr && node->right == nullptr) 
            ans += '(';
            dfs(node->left);
            ans += ')';
        
        // 只有右子树在
        if(node->left == nullptr && node->right != nullptr) 
            ans += "()(";
            dfs(node->right);
            ans += ')';
        
    
;