Codeforces 451E Devu and Flowers 组合数+容斥原理

http://codeforces.com/problemset/problem/451/E

题目大意：有n个花坛，要选s支花，每个花坛有f[i]支花。同一个花坛的花颜色相同，不同花坛的花颜色不同，问说可以有多少种组合。

解题思路：2n的状态，二进制枚举出那些花坛的花取超过了，剩下的用C(n−1sum+n−1)隔板法计算个数，注意奇数的位置要用减的，偶数的位置用加的，容斥原理。 
 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
ll f[30];
ll quick_mod(ll a, ll b, ll c)  
{
	ll ans = 1;
	while (b)
	{
		if (b % 2 == 1)
			ans = (ans*a) % c;
		b /= 2;
		a = (a*a) % c;
	}
	return ans;
}
ll comb(ll n, ll k,ll p) {
	if (n < k)
		return 0;
	if (k > n - k)
		k = n - k;
	ll a = 1, b = 1;
	for (int i = 0; i < k; i++) 
	{
		a = a * (n - i) % p;
		b = b * (i + 1) % p;
	}
	return a * quick_mod(b, p - 2,p) % p;
}
ll Lucas(ll n, ll m, ll p){
        return m ? Lucas(n/p, m/p, p) * comb(n%p, m%p, p) % p : 1;
}
int main(){
	ll n,s;
	scanf("%lld%lld",&n,&s);
	for(int i=0;i<n;i++)
	{
		scanf("%lld",&f[i]);
	}
    ll ans=0;
    for(int i=0;i<(1<<n);i++)//二进制枚举 
    {
    	int flag=1;
    	ll ts=s;
    	for(int j=0;j<n;j++)
    	{
    		if((1<<j)&i)
    		{
    			flag=-flag;
    			ts-=f[j]+1;
			}
		}
		if(ts<0)
		continue;
		ans=(ans+flag*Lucas(ts+n-1,n-1,mod))%mod;
	}printf("%lld\n",(ans+mod)%mod);
    return 0;
}