zju 3638 Fruit Ninja

zju 3638 Fruit Ninja

这个题目其实就是经典的容斥原理的题目，由于n个水果的限制条件不一样，不然n就可以大到1000也能搞，这题n为15，我是直接利用二进制枚举搞的

要注意的是: 求解C(n,m )  的时候要用到逆元，由于p是素数直接可以 以用 a^(p-2) * a == 1 mod( p ) 欧拉定理来求逆元

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<queue>
#include<cmath>

using namespace std;

typedef long long ll;
const int mod=100000007;

ll quickmod(ll a,int n)
{
    ll ans=1;
    for(;n;n>>=1,a=a*a%mod)
      if(n&1) ans=ans*a%mod;
    return ans;
}
ll C(ll a,ll b)
{
    if(a<b||a<0||b<0) return 0;
    ll ret=1,ret1=1;
    for(int i=0;i<b;i++) ret=ret*(a-i)%mod,ret1=ret1*(i+1)%mod;
    ret=ret*quickmod(ret1,mod-2)%mod;
    return ret;
}

int main()
{
    int n,m,lim[20],d;
    char s1[150],s2[150],s3[150],s[1000];

    while(scanf("%d%d",&n,&m)==2&&n)
    {
         int nn=0;
         gets(s);
         while(1)
         {
             if(!gets(s)) break;
             if(strlen(s)<2) break;
             sscanf(s,"%s %s %s %d",s1,s2,s3,&d);
             if(s2[0]=='g') m-=(d+1);
             else lim[nn++]=d;
         }
        if(m<0){
            printf("0\n");continue;
         }
         ll ans=0;
         for(int i=0;i<(1<<nn);i++)
         {
             int bit=0,tot=0;
             for(int j=0;j<nn;j++)
               if(i&(1<<j)) bit++,tot+=lim[j];

             if(bit&1) ans-=C(m-tot+n-1,n-1);
             else ans+=C(m-tot+n-1,n-1);
             ans=(ans%mod+mod)%mod;
         }
         printf("%lld\n",ans);
    }
    return 0;
}