「TJOI2019」唱、跳、rap 和篮球

题目链接

题目分析

据说这是一道生成函数题

看到限制条件，我们首先想到的就是对有多少组讨论cxk的人进行容斥。然后就是求剩下的人随便放有多少种方法了。考虑现在每种剩(a,b,c,d)人，还需要排(n)人，那么方案数就是

[sum_{i=1}^asum_{j=1}^bsum_{k=1}^csum_{l=1}^d [i+j+k+l=n]{nchoose i}{n - ichoose j}{n-i-jchoose k}{n-i-j-kchoose l} ]

其中([])内表达式为真时，值为(1)，否则为(0)。

可以化简为

[sum_{i=1}^asum_{j=1}^bsum_{k=1}^c[0leqslant n-i-j-kleqslant d]{nchoose i}{n-ichoose j}{n-i-j choose k} ]

这个式子暴力求得话复杂度明显不对，他是(O(n^3))的。

考虑只枚举(i)和(j)。那么满足(0leqslant n-i-j-kleqslant d)的(k)显然是一个连续的区间，不妨记作([L,R])。那么对答案的贡献就是

[{nchoose i}{n-ichoose j}sum_{k=L}^R{n-i-jchoose k} ]

后面那个求和预处理加前缀和即可(O(1))求得。那么时间复杂度就降到了(O(n^2))。加上外面一层容斥，那么总时间复杂度就是(O(n^3))。

实现的时候注意边界条件！！！

参考程序

码风氪化

#include <bits/stdc++.h>
#define LL long long
using namespace std;

const LL Maxn = 1010;
const LL Maxa = 510;
const LL Mod = 998244353;
LL N, A[ 4 ], Fact[ Maxn ], Inv[ Maxn ], C[ Maxn ][ Maxn ], Sum[ Maxn ][ Maxn ], Ans;

void Exgcd( LL a, LL b, LL &x, LL &y ) { if( b == 0 ) { x = 1; y = 0; return; } Exgcd( b, a % b, y, x ); y -= a / b * x; return; }
LL inv( LL a ) { LL x, y; Exgcd( a, Mod, x, y ); if( x < 0 ) x += Mod; return x; }
LL S( LL n, LL l, LL r ) { LL Ans = Sum[ n ][ r ]; if( l ) Ans = ( Ans - Sum[ n ][ l - 1 ] + Mod ) % Mod; return Ans; }

LL Cal( LL Sta ) {
	LL Ans = 0; for( LL i = 0; i < 4; ++i ) A[ i ] -= Sta;
	for( LL i = 0; i <= A[ 0 ] && i <= N - Sta * 4; ++i )
		for( LL j = max( 0LL, N - Sta * 4 - A[ 2 ] - A[ 3 ] - i ); j <= A[ 1 ] && i + j <= N - Sta * 4; ++j ) {
			Ans = ( Ans + C[ N - Sta * 4 ][ i ] * C[ N - Sta * 4 - i ][ j ] % Mod * S( N - Sta * 4 - i - j, max( 0LL, N - Sta * 4 - i - j - A[ 3 ] ), min( N - Sta * 4 - i - j, A[ 2 ] ) ) % Mod ) % Mod;
		}
	for( LL i = 0; i < 4; ++i ) A[ i ] += Sta; return Ans;
}
void Work() {
	for( LL i = 0; i <= N / 4; ++i ) {
		for( LL j = 0; j < 4; ++j ) if( i > A[ j ] ) return;
		if( i & 1 ) Ans = ( Ans - C[ N - i * 3 ][ i ] * Cal( i ) % Mod + Mod ) % Mod;
		else Ans = ( Ans + C[ N - i * 3 ][ i ] * Cal( i ) % Mod ) % Mod;
	}
	return;
}

int main() {
	scanf( "%lld", &N ); for( LL i = 0; i < 4; ++i ) scanf( "%lld", &A[ i ] );
	Fact[ 0 ] = 1; for( LL i = 1; i <= N; ++i ) Fact[ i ] = Fact[ i - 1 ] * i % Mod;
	Inv[ N ] = inv( Fact[ N ] ); for( LL i = N - 1; i >= 0; --i ) Inv[ i ] = Inv[ i + 1 ] * ( i + 1 ) % Mod;
	for( LL i = 0; i <= N; ++i ) for( LL j = 0; j <= i; ++j ) C[ i ][ j ] = Fact[ i ] * Inv[ j ] % Mod * Inv[ i - j ] % Mod;
	for( LL i = 0; i <= N; ++i ) for( LL j = 0; j <= i; ++j ) Sum[ i ][ j ] = ( Sum[ i ][ j - 1 ] + C[ i ][ j ] ) % Mod;
	Work(); printf( "%lld
", Ans );
	return 0;
}