2021-08-09 Luogu P2822

杨辉三角递推公式+二维前缀和

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define N 2005
#define Pi 3.141591653589793238462
#define M(x) memset(x,0,sizeof(x))
#define mod 1000000007
int n,m,k;
int C[N][N],ans[N][N];
void init()
{
    for(int i=0;i<N;++i)C[i][0]=1;
    C[1][1]=1;
    for(int i=2;i<N;++i)
    {
        for(int j=1;j<=i;++j)
        {
            C[i][j]=(C[i-1][j]+C[i-1][j-1])%k;//统计C[i][j]为0的有多少个就可以了
        }
    }
    for(int i=2;i<=2000;i++){
        for(int j=1;j<=i;j++){
            ans[i][j]=ans[i-1][j]+ans[i][j-1]-ans[i-1][j-1]+(C[i][j]==0);//二维前缀和
        }
        ans[i][i+1]=ans[i][i];//下一层需要
    }
}
int main()
{
   ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
   int _=1;cin>>_>>k;
   init();
   while(_--)
   {
       cin>>n>>m;
       if(m>n)m=n;//题目没有说m<=n
       cout<<ans[n][m]<<'\n';
   }
   return 0;
}