使用卡特兰数来解决的问题
作者:Grey
原文地址:
通项公式
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的数量