LeetCode 95. 不同的二叉搜索树 II--动态规划+链表

95. 不同的二叉搜索树 II

给你一个整数 n ，请你生成并返回所有由 n 个节点组成且节点值从 1 到 n 互不相同的不同 二叉搜索树 。可以按 任意顺序 返回答案。

示例 1：

输入：n = 3
输出：[[1,null,2,null,3],[1,null,3,2],[2,1,3],[3,1,null,null,2],[3,2,null,1]]

示例 2：

输入：n = 1
输出：[[1]]

提示：

1 <= n <= 8

题解

和这个题目一样的，LeetCode 96. 不同的二叉搜索树–动态规划。

先固定形状，再利用中序遍历把具体的数字填上就行。

AC代码

/**
 * 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 {
public:
    vector<TreeNode*>dp[10];
    vector<TreeNode*>fun(int x,int y)
    {
        vector<TreeNode*> res;
        if(x == 0)
        {
            for(int i=0;i<dp[y].size();i++)
            {
                TreeNode *p = new TreeNode(0);
                p->left = NULL;
                p->right = dp[y][i];
                res.push_back(p);
            }
        }
        else if(y==0)
        {
            for(int i=0;i<dp[x].size();i++)
            {
                TreeNode *p = new TreeNode(0);
                p->left = dp[x][i];
                p->right = NULL;
                res.push_back(p);
            }
        }
        else
        {
            for(int i=0;i<dp[x].size();i++)
            {
                for(int j=0;j<dp[y].size();j++)
                {
                    TreeNode *p = new TreeNode(0);
                    p->left = dp[x][i];//p作为当前状态的根节点，把左子树和右子树连接起来
                    p->right = dp[y][j];
                    res.push_back(p);
                }
            }            
        }
        return res;
    }
    int num;
    TreeNode *get_num(TreeNode *head,TreeNode *head2)//根据儿诧搜索树的形状给每个节点赋值
    {
        if(head==NULL)return NULL;
        head2->left = new TreeNode(0);
        head2->right = new TreeNode(0);
        head2->left = get_num(head->left,head2->left);
        num ++;
        head2->val = num;
        head2->right = get_num(head->right,head2->right);
        return head2;
    }
    vector<TreeNode*>RES;
    vector<TreeNode*> generateTrees(int n) 
    {
        TreeNode *p = new TreeNode(0);
        dp[1].push_back(p);//初始化只有一个根节点
        
        for(int i=2;i<=n;i++)
        {
            for(int j=0;j<i;j++)
            {
                vector<TreeNode*> res = fun(j,i-j-1);
                for(int k=0;k<res.size();k++)
                dp[i].push_back(res[k]);
            }
        }
        for(int i=0;i<dp[n].size();i++)
        {
            num = 0;
            TreeNode *p = new TreeNode(0);
            p = get_num(dp[n][i],p);
            RES.push_back(p);
        }
        return RES;
    }
};