【NOIP2016】洛谷2282 组合数问题

题目描述

组合数表示的是从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行，每行一个整数代表答案。

因为k的因数只有一两个，所以可以把乘除变成对指数的加减计算。
其实一个更简单的方法是在模k意义下用加法递推。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int k,maxp;
int prm[3];
struct big
{
    int t[3];
    big operator * (const big &b) const
    {
        int i;
        big ret;
        for (i=0;i<maxp;i++)
          ret.t[i]=t[i]+b.t[i];
        return ret;
    }
    big operator * (const int &x) const
    {
        int i,tem=x;
        big ret;
        for (i=0;i<maxp;i++)
        {
            ret.t[i]=t[i];
            while (tem%prm[i]==0)
            {
                ret.t[i]++;
                tem/=prm[i];
            }
        }
        return ret;
    }
    big operator / (const big &b) const
    {
        int i;
        big ret;
        for (i=0;i<maxp;i++)
          ret.t[i]=t[i]-b.t[i];
        return ret;
    }
    bool operator % (const int &x) const
    {
        int i,t1=x;
        big t2;
        memcpy(t2.t,t,sizeof(t));
        for (i=0;i<maxp;i++)
          while (t1%prm[i]==0)
          {
            if (t2.t[i]>0) t2.t[i]--;
            else return 1;
            t1/=prm[i];
          }
        return 0;
    }
}fac[2010],c[2010][2010],one;
int ans[2010][2010],T,qm[10010],qn[10010],maxn,maxm;
void init()
{
    int i,t1;
    scanf("%d%d",&T,&k);
    t1=k;
    for (i=2;t1>1;i++)
      if (t1%i==0)
      {
        prm[maxp++]=i;
        while (t1%i==0) t1/=i;
      }
    for (i=1;i<=T;i++)
    {
        scanf("%d%d",&qn[i],&qm[i]);
        maxn=max(maxn,qn[i]);
        maxm=max(maxm,qm[i]);
    }
}
void solve()
{
    int i,j;
    for (i=1;i<=maxn||i<=maxm;i++)
      fac[i]=fac[i-1]*i;
    for (i=0;i<=maxn;i++)
      for (j=0;j<=maxm&&j<=i;j++)
        c[i][j]=fac[i]/fac[j]/fac[i-j];
    for (i=0;i<=maxn;i++)
      for (j=0;j<=maxm;j++)
      {
        ans[i+1][j+1]=ans[i+1][j]+ans[i][j+1]-ans[i][j];
        if (i>=j&&!(c[i][j]%k)) ans[i+1][j+1]++;
      }
}
int main()
{
    int i;
    init();
    solve();
    for (i=1;i<=T;i++)
      printf("%d\n",ans[qn[i]+1][qm[i]+1]);
}