P2822 组合数问题——前缀和优化，递推打表

题目描述

组合数 C（n，m） 表示的是从 n 个物品中选出 m 个物品的方案数。举个例子，从 (1,2,3) 三个物品中选择两个物品可以有 (1,2),(1,3),(2,3)这三种选择方法。根据组合数的定义，我们可以给出计算组合数 C(n,m)​ 的一般公式：

                                                

其中 n!=1×2×⋯×n;           特别地，定义 0!=1;

小葱想知道如果给定n,m,k,对于所有的 0≤i≤n,0≤j≤min(i,m).有多少对 c(i,j)满足 (i,j)是 k 的倍数。

输入输出格式

输入格式：

 

第一行有两个整数 t,k，其中 t 代表该测试点总共有多少组测试数据，k 的意义见问题描述。

接下来 t 行每行两个整数 n,m，其中 n,m 的意义见问题描述。

 

输出格式：

 

共 t 行，每行一个整数代表所有的 0≤i≤n,0≤j≤min(i,m) 中有多少对 (i,j) 满足c(i,j) 是 k 的倍数。

 

输入输出样例

输入样例#1： 

1 2
3 3

输出样例#1： 

1

输入样例#2： 

2 5
4 5
6 7

输出样例#2： 

0
7

 

思路：

       对于2000的n,m数据来说，求解组合数好的方法就是递归打表

       这里需要知道一个数学上排列组合的公式：c(n,m)=c(n-1,m)+c(n-1,m-1);

       又可以看作是杨辉三角

       1

        1       1

         1       2      1

          1       3      3       1

           1       4      6       4       1

          ········

           每一项对应了组合数c(i,j)

          

            伪代码：

 

void init()
{
    c[0][0]=1;c[1][0]=1;c[1][1]=1;
    for(i=2;i<=2000;i++)
        for(j=0;j<=i;j++)
        {
           if(j==0)
                c[i][j]=1;
           else{
            c[i][j]=(c[i-1][j]+c[i-1][j-1]);
           }
        }
}

 

          对于t次询问，可以利用前缀和实现从o(n)的复杂度降为o(1)的复杂度

          //     学习前缀和推荐博客：https://www.cnblogs.com/mrclr/p/8423136.html

         

给定一个n*n的矩阵，找一个最大的子矩阵，使得这个子矩阵里面的元素和最大。

 

再想想前缀和矩阵怎么生成？看个图就明白了：

那么最大的矩形前缀和就等于蓝的矩阵加上绿的矩阵，再减去重叠面积，最后加上小方块，即

sum[i][j] = sum[i][j - 1] + sum[i - 1][j] - sum[i - 1][j - 1] + a[i][j]

 

对于这道题来说，利用前缀和需要注意两点

1：当c[i][j]%k==0，说明符合是k的倍数的个数增加了一个，所以前缀和数量加一，即a[i][j]++

2:对于a[i][i+1]=a[i]来说，是对每一行最后一个点的下一个点的前缀和进行更新，为的是在求下一行最后一个点时需要用到这个点。

 

 

虚线框就是对a[i][i+1]的赋值，来求虚线框正下方的值（蓝色表示组合数（杨辉三角），红色表示前缀和））

对于红色第一列为零，因为在代码中当j==0，就只更新c[i][j]=0，就不考虑了，然后第五行的2,3,4增加是因为在对应的组合数中对应的数分别是4，6，4，都是2的倍数，这就解释了代码中为什么if(c[i][j]==0)    a[i][j]++；的原因了

 

 

 

 

 

完整代码：

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cmath>
const int xmax=2050;
typedef long long ll;
using namespace std;

ll c[xmax][xmax],a[xmax][xmax];
int i,j,temp;
int t,k,n,m;

void init()
{
    c[0][0]=1;c[1][0]=1;c[1][1]=1;
    for(i=2;i<=2000;i++)
        for(j=0;j<=i;j++)
        {
           if(j==0)
                c[i][j]=1;
           else{
            //取模优化，是求解这道题的关键之一
            c[i][j]=(c[i-1][j]+c[i-1][j-1])%k;
            a[i][j]=a[i-1][j]+a[i][j-1]-a[i-1][j-1];
            if(c[i][j]==0)  a[i][j]++;
           }
           a[i][i+1]=a[i][i];
        }
}
int main()
{
    scanf("%d%d",&t,&k);
    init();
    while(t--)
    {
        scanf("%d%d",&n,&m);
        m=min(n,m);
        printf("%lld\n",a[n][m]);
    }
    return 0;
}