CodeForces451E Devu and Flowers

题目链接

问题分析

没有想到母函数的做法……

其实直接看题思路挺简单的。发现如果每种花都有无限多的话，问题变得十分简单，答案就是\(s+n-1\choose n - 1\)。然后发现\(n\)只有\(20\)，于是大力容斥一波就完事了。

参考代码

#include <cstdio>

const long long Max_n = 30;
const long long Mod = 1000000007;
long long n, s, f[ Max_n ];

void Exgcd( long long a, long long b, long long & x, long long & y ) {
	if( b == 0LL ) { x = 1LL; y = 0LL; return; }
	Exgcd( b, a % b, y, x );
	y -= a / b * x;
	return;
}

long long Inv( long long a ) {
	long long x, y;
	Exgcd( a, Mod, x, y );
	if( x < 0 ) x += Mod;
	return x;
}

long long C( long long n, long long m ) {
	long long Ans = 1;
	for( long long i = 1; i <= m; ++i ) Ans = Ans * ( ( n - i + 1 ) % Mod ) % Mod;
	for( long long i = 1; i <= m; ++i ) Ans = Ans * Inv( i ) % Mod;
	return Ans;
}

int main() {
	scanf( "%lld%lld", &n, &s );
	for( long long i = 1; i <= n; ++i ) scanf( "%lld", &f[ i ] );
	long long Ans = 0;
	for( long long i = 0; i < 1 << n; ++i ) {
		long long t, Cnt = 0, Pos = s;
		for( t = i; t; t >>= 1 ) if( t & 1 ) ++Cnt;
		for( long long j = 1, t = i; t; t >>= 1, ++j ) if( t & 1 ) Pos -= f[ j ] + 1;
		if( Pos < 0 ) continue;
		Ans += ( Cnt & 1 ) ? -C( Pos + n - 1, n - 1 ) : C( Pos + n - 1, n - 1 );
		Ans = ( Ans + Mod ) % Mod;
	}
	printf( "%lld\n", Ans );
	return 0;
}