Leetcode:96.不同的二叉搜索树&&Leetcode:95.不同的二叉搜索树II

最新推荐文章于 2022-02-25 23:47:59 发布

Liao_Hong_XiDian

最新推荐文章于 2022-02-25 23:47:59 发布

阅读量272

点赞数

分类专栏： Leetcode C 文章标签： Leetcode C 不同的二叉搜索树

本文链接：https://blog.csdn.net/qq_23523409/article/details/83997649

版权

Leetcode 同时被 2 个专栏收录

152 篇文章 0 订阅

订阅专栏

133 篇文章 0 订阅

订阅专栏

Leetcode:96.不同的二叉搜索树

给定一个整数 n，求以 1 ... n 为节点组成的二叉搜索树有多少种？

示例:

输入: 3
输出: 5
解释:
给定 n = 3, 一共有 5 种不同结构的二叉搜索树:

   1         3     3      2      1
    \       /     /      / \      \
     3     2     1      1   3      2
    /     /       \                 \
   2     1         2                 3

解题思路：

核心思想:卡特兰数列，动态规划（DP）。

f(n)=f(0)*f(n-1)+f(1)*f(n-2)+f(2)*f(n-3)+......+f(n-1)*f(0);

很显然，如果不用数组保存已经算过的值，就要重复计算多次相同的结果，当n非常大时，重复的量也非常多，难免超时。

C++代码
class Solution { public: int numTrees(int n) { if (n == 0) return 1; if (n == 1) return 1; if (data[n] != 0) return data[n]; int sum = 0; for (int i = 1; i <= n; i++) { sum += numTrees(i - 1)*numTrees(n - i); } data[n] = sum; return sum; } private: vector<int> data = vector<int>(10000, 0); };

Leetcode:95.不同的二叉搜索树II

核心思想：卡特兰数列，递归，动态规划。先按照递归的思想生成n个结点所有形状的树，之后再将1-n按照中序遍历的顺序写入树中。

根据这题，本人编写了如下的关于二叉树的几个功能。

1. 树的复制。复制一个二叉树，形状与root相同，但是重新分配了内存。

TreeNode* copyTree(TreeNode* root)；

2. 查找n个结点的树的形状。将他们的根结点储存起来，确保这些树没有共同的结点。

vector<TreeNode*> find(int n)；

3. 按照中序遍历的顺序，将1-n写入一个二叉树中。

void In_order(TreeNode* root)；

这个代码最快速度为16ms，多次出现，相比8ms，12ms的代码而言，我觉得主要的差距在于，我这个生成树的形状之后，将所有结点赋值为0，最后又要按照中序遍历顺序将1-n赋值到结点中，这里出现了多余的2倍操作，但是按照这个思路不太好一开始将应有的数值赋值，后续仍需改进。

C++代码
#define hasLChild(x) (!(x->left==NULL)) #define hasRChild(x) (!(x->right==NULL)) class Solution { public: vector<TreeNode> generateTrees(int n) { if (n <= 0) return{}; vector<TreeNode> res = find(n); //将中序遍历所有二叉树，将1-n写入树中 for (int i = 1; i <= int(res.size()); i++) { num = 1; In_order(res[i - 1]); } return res; } void In_order(TreeNode* root) { if (hasLChild(root)) In_order(root->left); root->val = num; num++; if (hasRChild(root)) In_order(root->right); } vector<TreeNode> find(int n) { if (n == 0) return{ NULL }; if (n == 1) return{ new TreeNode(0) }; if (int(dp[n].size()) > 0) return dp[n]; vector<TreeNode> res; vector<TreeNode> data1, data2; for (int i = 1; i <= n; i++) { data1 = find(i - 1); data2 = find(n - i); for (int j = 1; j <= int(data1.size()); j++) { for (int k = 1; k <= int(data2.size()); k++) { TreeNode temp = new TreeNode(0); temp->right = copyTree(data1[j - 1]); temp->left = copyTree(data2[k - 1]); res.push_back(temp); } } } dp[n] = res; return res; } TreeNode* copyTree(TreeNode* root) {//将root复制到copy中 TreeNode* copy; if (root == NULL) { copy = NULL; return copy; } copy = new TreeNode(0); queue<TreeNode> Qr, Qc; TreeNode tempr,tempc; Qr.push(root); Qc.push(copy); while (!Qr.empty()) { tempr = Qr.front(); tempc = Qc.front(); Qr.pop(); Qc.pop(); if (hasLChild(tempr)) { tempc->left = new TreeNode(0); Qr.push(tempr->left); Qc.push(tempc->left); } if (hasRChild(tempr)) { tempc->right = new TreeNode(0); Qr.push(tempr->right); Qc.push(tempc->right); } } return copy; } private: int num = 0; vector<TreeNode> dp[1000]; };

参考的别人的更优秀的代码，学到了很多，平均用时12ms，如下。

C++代码
class Solution { public: vector<TreeNode> generateTrees(int n) { if(n==0) return vector<TreeNode>(); return dfs_genTree(1,n); } vector<TreeNode> dfs_genTree(int b,int e){ vector<TreeNode> res; if(b>e){ res.push_back(NULL); return res; } for(int i=b;i<=e;i++){ vector<TreeNode> l = dfs_genTree(b,i-1); vector<TreeNode> r = dfs_genTree(i+1,e); for(auto j:l){ for(auto k:r){ TreeNode* tn = new TreeNode(i); tn->left = j; tn->right = k; res.push_back(tn); } } } return res; } };