47

方法 1:
C(n,r) = n!/(nr)!r!

方法 2:
wilf 的《组合算法》一书中,我发现:
C(n,r) 可以写为C(n-1,r) + C(n-1,r-1).

例如

C(7,4) = C(6,4) + C(6,3) 
       = C(5,4) + C(5,3) + C(5,3) + C(5,2)
       .   .
       .   .
       .   .
       .   .
       After solving
       = C(4,4) + C(4,1) + 3*C(3,3) + 3*C(3,1) + 6*C(2,1) + 6*C(2,2)

如您所见,最终的解决方案不需要任何乘法。在每种形式 C(n,r) 中,n==r 或 r==1。

这是我实现的示例代码:

int foo(int n,int r)
{
     if(n==r) return 1;
     if(r==1) return n;
     return foo(n-1,r) + foo(n-1,r-1);
}

请参阅此处的输出

在方法 2 中,存在重叠的子问题,我们调用递归来再次解决相同的子问题。我们可以通过使用动态规划来避免它。

我想知道计算 C(n,r) 的更好方法是什么?

4

5 回答 5

84

两种方法都会节省时间,但第一种方法很容易出现整数溢出

方法一:

这种方法将在最短的时间内(最多n/2迭代)产生结果,并且可以通过仔细进行乘法来减少溢出的可能性:

long long C(int n, int r) {
    if(r > n - r) r = n - r; // because C(n, r) == C(n, n - r)
    long long ans = 1;
    int i;

    for(i = 1; i <= r; i++) {
        ans *= n - r + i;
        ans /= i;
    }

    return ans;
}

此代码将从较小的一端开始乘分子,并且由于任何k连续整数的乘积都可以被 整除k!,因此不会有可除性问题。但是溢出的可能性仍然存在,另一个有用的技巧可能是在进行乘法和除法之前除以它们的 GCD(并且n - r + i仍然可能发生溢出)。i

方法二:

在这种方法中,您实际上将构建帕斯卡三角。动态方法比递归方法快得多(第一个是指数的O(n^2),而另一个是指数的)。但是,您也需要使用O(n^2)内存。

# define MAX 100 // assuming we need first 100 rows
long long triangle[MAX + 1][MAX + 1];

void makeTriangle() {
    int i, j;

    // initialize the first row
    triangle[0][0] = 1; // C(0, 0) = 1

    for(i = 1; i < MAX; i++) {
        triangle[i][0] = 1; // C(i, 0) = 1
        for(j = 1; j <= i; j++) {
            triangle[i][j] = triangle[i - 1][j - 1] + triangle[i - 1][j];
        }
    }
}

long long C(int n, int r) {
    return triangle[n][r];
}

然后,您可以随时查找C(n, r)O(1)

如果您需要特定的C(n, r)(即不需要完整的三角形),则可以O(n)通过从上到下覆盖三角形的同一行来消耗内存。

# define MAX 100
long long row[MAX + 1];

int C(int n, int r) {
    int i, j;

    // initialize by the first row
    row[0] = 1; // this is the value of C(0, 0)

    for(i = 1; i <= n; i++) {
        for(j = i; j > 0; j--) {
             // from the recurrence C(n, r) = C(n - 1, r - 1) + C(n - 1, r)
             row[j] += row[j - 1];
        }
    }

    return row[r];
}

内循环从头开始,以简化计算。如果从索引 0 开始,则需要另一个变量来存储被覆盖的值。

于 2012-08-04T15:32:42.213 回答
11

我认为您的递归方法应该有效地与DP. 但是一旦约束增加,它就会开始出现问题。见http://www.spoj.pl/problems/MARBLES/

这是我在在线评委和编码比赛中使用的功能。所以它的工作速度非常快。

long combi(int n,int k)
{
    long ans=1;
    k=k>n-k?n-k:k;
    int j=1;
    for(;j<=k;j++,n--)
    {
        if(n%j==0)
        {
            ans*=n/j;
        }else
        if(ans%j==0)
        {
            ans=ans/j*n;
        }else
        {
            ans=(ans*n)/j;
        }
    }
    return ans;
}

这是您的方法 #1 的有效实现

于 2012-08-04T14:58:22.977 回答
7

您的递归方法很好,但是在您的方法中使用 DP 将减少再次解决子问题的开销。现在因为我们已经有两个条件-

nCr(n,r) = nCr(n-1,r-1) + nCr(n-1,r);

nCr(n,0)=nCr(n,n)=1;

现在我们可以通过将子结果存储在二维数组中来轻松构建 DP 解决方案 -

int dp[max][max];
//Initialise array elements with zero
int nCr(int n, int r)
{
       if(n==r) return dp[n][r] = 1; //Base Case
       if(r==0) return dp[n][r] = 1; //Base Case
       if(r==1) return dp[n][r] = n;
       if(dp[n][r]) return dp[n][r]; // Using Subproblem Result
       return dp[n][r] = nCr(n-1,r) + nCr(n-1,r-1);
}

现在,如果您想进一步优化,获取二项式系数的素数分解可能是计算它的最有效方法,尤其是在乘法成本高昂的情况下。

我知道最快的方法是弗拉基米尔的方法。一种方法是通过将 nCr 分解为素因子来避免全部除法。正如弗拉基米尔所说,您可以使用 Eratosthenes 筛子非常有效地做到这一点。此外,使用费马小定理计算 nCr mod MOD(其中 MOD 是质数)。

于 2016-06-05T12:50:01.683 回答
0

使用动态编程,您可以轻松找到 nCr 这里是解决方案

package com.practice.competitive.maths;

import java.util.Scanner;

public class NCR1 {

    public static void main(String[] args) {
        try (Scanner scanner = new Scanner(System.in)) {
            int testCase = scanner.nextInt();
            while (testCase-- > 0) {
                int n = scanner.nextInt();
                int r = scanner.nextInt();
                int[][] combination = combination();
                System.out.println(combination[n][r]%1000000007);
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static int[][] combination() {
        int combination[][] = new int[1001][1001];
        for (int i = 0; i < 1001; i++)
            for (int j = 0; j <= i; j++) {
                if (j == 0 || j == i)
                    combination[i][j] = 1;
                else
                    combination[i][j] = combination[i - 1][j - 1] % 1000000007 + combination[i - 1][j] % 1000000007;
            }
        return combination;
    }
}
于 2019-01-03T09:49:31.457 回答
-1
unsigned long long ans = 1,a=1,b=1;
        int k = r,i=0;
        if (r > (n-r))
            k = n-r;
        for (i = n ; k >=1 ; k--,i--)
        {
            a *= i;
            b *= k;
            if (a%b == 0)
            {
                a = (a/b);
                b=1;
            }
        }
        ans = a/b;
于 2014-11-18T07:24:26.350 回答