https://www.luogu.org/problem/show?pid=2822
题目描述
组合数表示的是从n个物品中选出m个物品的方案数。举个例子,从(1,2,3) 三个物品中选择两个物品可以有(1,2),(1,3),(2,3)这三种选择方法。根据组合数的定 义,我们可以给出计算组合数的一般公式:
其中n! = 1 × 2 × · · · × n
小葱想知道如果给定n,m和k,对于所有的0 <= i <= n,0 <= j <= min(i,m)有多少对 (i,j)满足是k的倍数。
输入输出格式
输入格式:
第一行有两个整数t,k,其中t代表该测试点总共有多少组测试数据,k的意义见 【问题描述】。
接下来t行每行两个整数n,m,其中n,m的意义见【问题描述】。
输出格式:
t行,每行一个整数代表答案。
输入输出样例
输入样例#1:
1 2
3 3
输出样例#1:
1
输入样例#2:
2 5
4 5
6 7
输出样例#2:
0
7
说明
【样例1说明】
在所有可能的情况中,只有是2的倍数。
【子任务】
递推+二维前缀和
组合数递推公式:C(i,j)=C(i-1,j)+C(i-1,j-1)
预处理所有组合数%k是否为0,注意递推过程中取模,同时标记%k=0的位置
前缀和统计答案 sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+(模k是否为0)
注意特殊处理0
n²预处理 O(1)查询
#include<cstdio>
#include<cstring>
using namespace std;
int t,k,c[][],sum[][],a[][];
void pre()
{
for(int i=;i<=;i++)
{
c[i][]=%k,c[i][]=i%k;
if(!c[i][]) a[i][]++;
if(!c[i][]) a[i][]++;
}
for(int i=;i<=;i++)
for(int j=;j<=i;j++)
{
c[i][j]=(c[i-][j]+c[i-][j-])%k;
if(!c[i][j]) a[i][j]++;
}
for(int i=;i<=;i++) sum[i][]=sum[i-][]+a[i][],sum[][i]=sum[][i-]+a[][i];
for(int i=;i<=;i++)
{
for(int j=;j<=;j++)
sum[i][j]=sum[i-][j]+sum[i][j-]-sum[i-][j-]+a[i][j];
} }
int main()
{
freopen("problem.in","r",stdin);
freopen("problem.out","w",stdout);
scanf("%d%d",&t,&k);
pre();
int n,m;
while(t--)
{
scanf("%d%d",&n,&m);
printf("%d\n",sum[n][m]);
}
return ;
}
两个错误:
1、sum数组没有预处理0,sum[][0-1]出问题了
2、递推过程中第二维j循环到i,不能到2000,对C数组没有影响,
但对a数组来说,c未经赋值为0,错误统计