【C++】【LeetCode】96. Unique Binary Search Trees

xiaoxiao2021-02-28  83

题目

Given n, how many structurally unique BST’s (binary search trees) that store values 1…n? For example, Given n = 3, there are a total of 5 unique BST’s.

思路

给定n,求不重复二叉查找树的数量,可以分为n种情况,根节点是x(x的取值从1到n),左边都是比x小的数,右边都是比x大的数。左右两边又可以递归遍历根节点,求出在根节点为该数的情况下的不重复二叉查找树的数量,当左边的不重复二叉查找树的数量为y,右边为z时,该根节点为该数的不重复二叉查找树的数量为y*z。

代码

class Solution { public: int numTrees(int n) { vector<int> data(n+1); data[0] = 1; data[1] = 1; helper(data, n); return data[n]; } void helper(vector<int>& data, int n) { int count = 0; for (int i = 0; i <= n - 1; i++) { if (data[max(n - 1, i)] == 0) { helper(data, max(n - 1 - i, i)); } count += data[i] * data[(n - 1 - i)]; } data[n] = count; } };
转载请注明原文地址: https://www.6miu.com/read-65098.html

最新回复(0)