AtCoder Grand Contest 025 - B - RGB Coloring

题目

题意：

给你一座N层的塔，让你给每一层塔染色或者不染。给你A、B

染红色加A，绿色加A+B，蓝色加B。

 

POINT：

可以先不考虑绿色。假如只有红色和蓝色。我们遍历红色的个数从0到N。

算出对应的蓝色数，得到红色X，蓝色Y。那么方案数为C(X,N)*C(Y,N)。

然后独立的涂色，肯定会有涂重的部分。那么这个涂了红色又涂了蓝色的部分就是涂了绿色的部分。

 

#include <iostream>
#include <stdio.h>
using namespace std;
#define LL long long
const LL maxn = 3e5+55;
const LL mod = 998244353;
LL jie[maxn];

LL qkm(LL base,LL mi)
{
	LL ans=1;
	while(mi){
		if(mi&1)
			ans=ans*base%mod;
		base=base*base%mod;
		mi>>=1;
	}
	return ans;
}

LL C(LL a,LL b)
{
	//printf("%lld %lld %lld\n",jie[b],qkm(jie[b-a],mod-2),qkm(jie[a],mod-2));
	return jie[b]*qkm(jie[b-a],mod-2)%mod*qkm(jie[a],mod-2)%mod;
}


int main()
{



	jie[0]=1;
	for(LL i=1;i<maxn;i++)
		jie[i]=jie[i-1]*i%mod;


	LL n,a,b,k;
	scanf("%lld%lld%lld%lld",&n,&a,&b,&k);
	LL ans=0;
	for(LL i=0;i<=n;i++){
		if(k-i*a<0) break;
		if((k-i*a)%b==0){
			LL y=(k-i*a)/b;
			if(y<=n){
				ans+=C(i,n)*C(y,n)%mod;
				ans%=mod;
			}
		}
	}
	printf("%lld\n",ans);
	return 0;
}