96. 不同的二叉搜索树

96. 不同的二叉搜索树_第1张图片

力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台

96. 不同的二叉搜索树_第2张图片

96. 不同的二叉搜索树_第3张图片

96. 不同的二叉搜索树_第4张图片

class Solution {
public:
    int numTrees(int n) {
        std::vector visited(n+1, -1);
        return dfs(n, visited);
    }
private:
    int dfs(int n, std::vector& visited) {
        if (n == 0 || n == 1) {
            return 1;
        }
        if (visited[n] != -1) {
            return visited[n];
        }
        int result = 0;
        for (int i = 1; i <= n; ++i) {
            int tmp = dfs(i-1, visited) * dfs(n-i, visited);
            result += tmp;
        }
        visited[n] = result;
        return result;
    }
};

你可能感兴趣的:(leetcode,深度优先,算法)