day2 T1暴力打残只得了20分,出考场才知道是个杨辉三角,悔恨莫及。
组合数公式
Cij=s[i−1][j−1]+s[i−1][j]
,这个应该不难理解,2000*2000的暴力算一下就可以了,然后二维前缀和维护,注意,杨辉三角不是矩形,当计算sum[i][i]时,会用到sum[i-1][i],而在杨辉三角中没有列大于行的情况,所以在计算的时候要
sum[i][i+1]=sum[i][i]
,否则会出错。
最后
O(1)
查询就OK了
代码如下
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int n,k,sum[2017][2017],s[2017][2017],a,b,i,j;
void read(int &x)
{
x=0;char c = getchar();
while(c<48||c>57) c=getchar();
while(c>=48&&c<=57) x=x*10+c-48 ,c=getchar();
return ;
}
int main()
{
read(n);read(k);
s[0][0]=1;
for(i=1;i<=2005;i++)
{
s[i][0]=1;
for(j=1;j<=i;j++)
{
sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1];
s[i][j]=(s[i-1][j]+s[i-1][j-1])%k;
if(s[i][j]==0) sum[i][j]++;
}
sum[i][i+1]=sum[i][i];
}
for(i=1;i<=n;i++)
{
read(a);read(b);
printf("%d\n",sum[a][min(a,b)]) ;
}
return 0;
}