使用卡特兰数来解决的问题

时间:2022-09-27 20:06:24

作者:Grey

原文地址:

博客园:使用卡特兰数来解决的问题

CSDN:使用卡特兰数来解决的问题

通项公式

k(0) = 1, k(1) = 1,如果接下来的项满足:

k(n) = k(0) x k(n - 1) + k(1) x k(n - 2) + …… + k(n - 2) x k(1) + k(n - 1) x k(0)

或者

k(n) = C(2n, n) - C(2n, n-1)

或者

k(n) = C(2n, n) / (n + 1)

就说这个表达式,满足卡特兰数。

比如

n 个左括号,n 个右括号,有多少种合法的组合方式?合法的定义是任何一个前缀串,右括号的数量必须小于左括号的数量

合法的不好求,我们可以先求不合法的,因为总的方法数是C(2n,n)(先安排 n 个左括号,另外的位置自然成为右括号的位置)

不合法的情况是:一定存在一个前缀,右括号的数量 = 左括号的数量 + 1,即不合法的数量等于C(2n, n+1),

所以合法的数量等于C(2n,n) - C(2n,n+1),即C(2n,n) - C(2n,n-1)

满足卡特兰数。

再如

给定 n 个数字,且每个数字都必须入栈,也必须出栈,求这些数合法的出栈入栈的顺序有多少种?

由于每个数字有出栈和入栈两个操作,所以,一共的操作组合有(包括不合法的方式)C(2n,n),

由于出栈的次数一定不可能大于入栈的次数,所以,不合法的组合方式中:一定存在一个出入栈的方式,出栈的次数 = 入栈次数 + 1,即C(2n, n + 1),合法的出入栈次数是C(2n,n) - C(2n, n + 1),即C(2n, n) - C(2n, n - 1),满足卡特兰数。

类似的还有

曲线在第一象限,可上升,可下降,求有多少种组合方式?

也满足卡特兰数。

N个节点有多少种形态的二叉树

有N个二叉树节点,每个节点彼此之间无任何差别,返回由N个二叉树节点,组成的不同结构数量是多少?

题目链接:

LintCode 163 · Unique Binary Search Trees

LeetCode 96. Unique Binary Search Trees

主要思路

有 0 个节点的时候,只有 1 种方法,即空树

有 1 个节点的时候,只有 1 种方法,即只有一个节点的树

有 2 个节点的时候,有 2 种方法,分别是

使用卡特兰数来解决的问题

即: k(0) = 1, k(1) = 1, k(2) = 2

当数量为 n 时,有如下一些情况,根节点占一个节点,然后

左树 0 个节点 ,右数 n - 1 个节点;

左树 1 个节点,右数 n - 2 个节点;

左树 2 个节点,右数 n - 3 个节点;
……
左树 n - 1 个节点 ,右数 0 个节点;

左树 n - 2 个节点,右数 1 个节点;

左树 n - 3 个节点,右数 2 个节点;

即:k(n) = k(0) x k(n - 1) + k(1) x k(n - 2) + …… + k(n - 2) x k(1) + k(n - 1) x k(0),满足卡特兰数。

完整代码如下

import java.math.BigInteger;
public class Solution {
    /**
     * @param n: An integer
     * @return: An integer
     */
    public static int numTrees(int n) {
        if (n < 0) {
            return BigInteger.ZERO.intValue();
        }
        if (n < 2) {
            return BigInteger.ONE.intValue();
        }
        BigInteger a = BigInteger.ONE;
        BigInteger b = BigInteger.ONE;
        for (int i = 1, j = n + 1; i <= n; i++, j++) {
            a = a.multiply(BigInteger.valueOf(i));
            b = b.multiply(BigInteger.valueOf(j));
            BigInteger gcd = gcd(a, b);
            a = a.divide(gcd);
            b = b.divide(gcd);
        }
        return (b.divide(a)).divide(BigInteger.valueOf(n + 1)).intValue();
    }

    public static BigInteger gcd(BigInteger m, BigInteger n) {
        return n.equals(BigInteger.ZERO) ? m : gcd(n, m.mod(n));
    }

    private static int numTrees2(int n) {
        if (n < 0) {
            return BigInteger.ZERO.intValue();
        }
        if (n < 2) {
            return BigInteger.ONE.intValue();
        }
        BigInteger a = BigInteger.valueOf(n + 1);
        BigInteger b = BigInteger.valueOf(1);
        for (int i = n + 2; i <= (2 * n); i++) {
            a = a.multiply(BigInteger.valueOf(i));
        }
        for (int i = 1; i <= n; i++) {
            b = b.multiply(BigInteger.valueOf(i));
        }
        return a.divide(b).divide(BigInteger.valueOf(n + 1)).intValue();
    }
}

1 0 前缀串数量问题

假设给你 n 个 0 和 n 个 1,你必须用全部数字拼序列,返回有多少个序列满足:任何前缀串,1 的数量都不少于 0 的数量

n 个 1 和 n 个 0,所有的排列组合是C(2n,n),由于合法数量 = 所有组合 - 非法数量,即

C(2n,n) - C(2n,n-1)

完整代码如下

package snippet;

import java.util.*;

//假设给你N个0,和N个1,你必须用全部数字拼序列
// 返回有多少个序列满足:任何前缀串,1的数量都不少于0的数量
// 卡特兰数
public class Code_10Ways {

    public static long ways2(int N) {
        if (N < 0) {
            return 0;
        }
        if (N < 2) {
            return 1;
        }
        long a = 1;
        long b = 1;
        long limit = N << 1;
        for (long i = 1; i <= limit; i++) {
            if (i <= N) {
                a *= i;
            } else {
                b *= i;
            }
        }
        return (b / a) / (N + 1);
    }
}

类似的问题

偶数(2N)个人排队,排两行,任何一个排在后面的人都不能比排在前面的人小,有几种排列方式?

其本质就是:前面N个人编号成0,后面N个人编号成1,任何前缀串,1的数量不小于0的数量

更多

算法和数据结构笔记

参考资料

算法和数据结构体系班-左程云