[提高][NOIP2016 提高组] 组合数问题

题目背景

NOIP2016 提高组 D2T1

题目描述

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

(��)=�!�!(�−�)!(mn​)=m!(n−m)!n!​

其中 �!=1×2×⋯×�n!=1×2×⋯×n；特别地，定义 0!=10!=1。

小葱想知道如果给定 �,�n,m 和 �k，对于所有的 0≤�≤�,0≤�≤min⁡(�,�)0≤i≤n,0≤j≤min(i,m) 有多少对 (�,�)(i,j) 满足 �∣(��)k∣(ji​)。

也就是(��)(ji​)是�k的倍数

输入格式

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

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

输出格式

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

输入数据 1

1 2
3 3

Copy

输出数据 1

1

Copy

输入数据 2

2 5
4 5
6 7

Copy

输出数据 2

0
7

Copy

数据规模与约定

题解

数据范围不大，因此直接预处理出 20002000 以内的组合数，运用杨辉三角，同时对 �k 取模，可以边计算边取模，否则会超出存储范围。

对于每一次输入的 �n 和 �m，都要从第 00 层开始求，频繁的求和，因此可以用二维前缀和预处理。二维前缀和用的时候保证是一个完整的矩阵，但是杨辉三角显然不是，是一个三角形形状，没关系我们给它没有数字的地方把数字补上，都补 11 就完事了,补成 11 预处理的时候是对 �k 取模的，对答案有贡献的都是取模后为 00 的我们补成 11 这样才不会影响最终的答案，can u understand?。

然后对于每一组询问，直接输出答案。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 2e3 + 5;
int t, k, c[N][N], sum[N][N];
void init()
{
    for (int i = 0; i <= 2000; i++)
    {
        for (int j = 0; j <= 2000; j++)
        {
            if (j > i) c[i][j] = c[i][j - 1];
            else
            {
                if (j == 0 || j == i) c[i][j] = 1;
                else c[i][j] = (c[i - 1][j] + c[i - 1][j - 1]) % k;
            }
        }
    }
    for (int i = 1; i <= 2000; i++)
    {
        for (int j = 1; j <= 2000; j++)
        {
            sum[i][j] = sum[i - 1][j] + sum[i][j - 1] - sum[i - 1][j - 1] + (c[i][j] == 0);
        }
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin >> t >> k;
    init();
    while (t--)
    {
        int n, m;
        cin >> n >> m;
        cout << sum[n][min(m, n)] << "\n";
    }
    return 0;
}