一、前言
分类:动态规划。
问题来源LeetCode 96 难度:中等。
问题链接:https://leetcode-cn.com/problems/unique-binary-search-trees/
二、题目
给定一个整数 n,生成所有由 1 ... n 为节点所组成的 二叉搜索树 。
示例:
输入:3
输出:5
解释:
以上的输出对应以下 5 种不同结构的二叉搜索树:
1 3 3 2 1
\ / / / \ \
3 2 1 1 3 2
/ / \ \
2 1 2 3
三、思路
这里提供两种解题方法:递归和非递归(动态规划)。思路和《95不同的二叉搜索树II》类似但是更简单。
四、编码实现
//==========================================================================
/*
* @file : 96_NumTrees.h
* @blogs : https://blog.csdn.net/nie2314550441/article/details/107215516
* @author : niebingyu
* @date : 2020/07/08
* @title : 96不同的二叉搜索树
* @purpose : 给定一个整数 n,求以 1 ... n 为节点组成的二叉搜索树有多少种?
*
* 示例:
* 输入:3
* 输出:5
* 解释:
* 以上的输出对应以下 5 种不同结构的二叉搜索树:
* 1 3 3 2 1
* \ / / / \ \
* 3 2 1 1 3 2
* / / \ \
* 2 1 2 3
*
*
* 来源:力扣(LeetCode)
* 难度:中等
* 链接:https://leetcode-cn.com/problems/unique-binary-search-trees/
*/
//==========================================================================
#pragma once
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
#define NAMESPACE_NUMTREES namespace NAME_NUMTREES {
#define NAMESPACE_NUMTREESEND }
NAMESPACE_NUMTREES
// 递归
class Solution_1
{
public:
int numTrees(int n)
{
if (n == 0)
return 0;
return helper(1, n);
}
int helper(int start, int end)
{
if (start > end)
return 1;
int sum = 0;
for (int i = start; i <= end; i++)
{
sum += helper(start, i - 1) * helper(i + 1, end);
}
return sum;
}
};
// 动态规划:非递归版本的拷贝树实现
// 递推公式:dp[n] = dp[i-1]*dp[n-i]; i=[1,n]。结果需要进行拷贝
class Solution_2
{
public:
int numTrees(int n)
{
if (n == 0)
return 0;
vector<int> dp(n + 1);
dp[0] = 1, dp[1] = 1;
for (int i = 2; i <= n; i++)
{
for (int j = 1; j <= i; j++)
{
dp[i] += dp[j - 1] * dp[i - j];
}
}
return dp[n];
}
};
以下为测试代码//
// 测试 用例 START
void test(const char* testName, int n, int expect)
{
Solution_1 s1;
int result1 = s1.numTrees(n);
Solution_2 s2;
int result2 = s2.numTrees(n);
if (expect == result1 && expect == result2) // 简单校验一下
cout << testName << ", solution passed." << endl;
else
cout << testName << ", solution failed. expect: "<< expect << ",result1: "<< result1 <<",result2:" << result2 << endl;
}
// 测试用例
void Test1()
{
int n = 1;
int expect = 1;
test("Test1()", n, expect);
}
void Test2()
{
int n = 2;
int expect = 2;
test("Test2()", n, expect);
}
void Test3()
{
int n = 3;
int expect = 5;
test("Test3()", n, expect);
}
void Test4()
{
int n = 4;
int expect = 14;
test("Test4()", n, expect);
}
NAMESPACE_NUMTREESEND
// 测试 用例 END
//
void NumTrees_Test()
{
NAME_NUMTREES::Test1();
NAME_NUMTREES::Test2();
NAME_NUMTREES::Test3();
NAME_NUMTREES::Test4();
}
执行结果: