hdu 3944 组合数学

哎，之前组合数学简直是弱的不能再差了。。从这道题中学到好多。。。

这题题意很简单，就是找一条如题意所说的路径，使数的总和最小，然后结果模一个素数。。要想找最小的方案，贪心即可，这里略掉。当k>n/2时，k=k-n/2，这样答案不会变。所以方案是SUM=n-k+C(n,k)+C(n-1,k-1)+...+C(n-k+1,0)。在由公式C(n,k)=C(n-1,k-1)+C(n-1,k)化简成SUM=n-k+C(n+1,k)....如何计算C(n+1,k)%p?根据lucas定理:C(n,k)=

C(n%p,k%p)*C(n/p,k/p)%p,其中当n<k时,C(n,k)=0...当n,k<p时，C(n,k)=n!*[(n-k)!k!]^-1%p...关于逆元的求法，就用扩展欧几里得定理。。这里略掉。还有，这道题由于样例极多，所以要事先预处理。。。附代码:

#include <iostream>
#include <vector>

using namespace std;

#define LL __int64
#define pb push_back

const int N=10005;
const int M=1234;

int c[M][N],a[N];
vector<int>b;
int p,m;

void extend(int a0,int b0,int &x,int &y)
{
    if (!b0)
    {
        x=1;y=0;
    }
    else
    {
        extend(b0,a0%b0,x,y);
        int temp=x;x=y;
        y=temp-a0/b0*y;
    }
}

LL js(int n,int k)
{
    LL res,d;
    int x=0,y=0;
    if (n<k) return 0;
    else if (n<p) 
    {
        d=c[m][n-k]*c[m][k]%p;
        extend(d,p,x,y);
        x=(x+p)%p;
        res=(c[m][n]*x)%p;
        return res;
    }
    else return js(n%p,k%p)*js(n/p,k/p)%p;
}

void pre()
{
    int i,j;
    for (j=0;j<b.size();j++)
    {
        LL r=1;
        c[j][0]=1;
        for (i=1;i<b[j];i++) 
        {
            r=(r*i)%b[j];
            c[j][i]=r;
        }    
    } 
}

int find(int l,int r)
{
    if (r-l<=1)
    {
        if (p==b[l]) return l;
        else return r;    
    }
    int t=(l+r)>>1;
    if (b[t]==p) return t;
    else if (b[t]>p) return find(l,t-1);
    else return find(t+1,r);
}

void prime()
{
    int i,j;
    for (i=2;i<N;i++)
        if (!a[i])
        {
            b.pb(i);
            for (j=i*2;j<N;j+=i) a[j]=1; 
        }
}

int main()
{
    int cas=0,n,k;
    LL sum;
    prime();
    pre();
    while (scanf("%d%d%d",&n,&k,&p)!=EOF)
    {
        cas++;
        if (k>n/2) k=n-k;
        if (k==0) sum=(n+1)%p;
        else 
        {
            m=find(0,b.size()-1);
            sum=(js(n+1,k)+n-k)%p;
        }
        printf("Case #%d: %d\n",cas,sum);        
    }
    return 0;
}