RGB Coloring（AGC-025）

题目链接：https://agc025.contest.atcoder.jp/tasks/agc025_b

题意：

给你一座塔有N层，让你涂3种颜色，分别是红色，绿色，蓝色。

给你A，B                A是红色的得分，B是蓝色的得分，A+B是绿色的得分。

给你目标分数K。

问题是：在给定N,A,B,K后，让你在N层塔涂上颜色，得到对应的分数K的方案数有多少种。

分析问题。

首先我们需要可以枚举N（< = 1e5 ）的层数，分别涂上红色。

然后看是否剩下来的得分能否被蓝色分数整除，若能整除且满足（蓝色格子的个数<=N）的情况。

答案就是  ans+=C(N,i)*C(N,(K-i*A)/B);

我们看看为什么可以这么做。我们在过程中完全忽略了绿色，

但其实没有，因为蓝色所覆盖的地方只要大于（N-红色格子）就相当于绿色所存在的地方。

贴上代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll mod = 998244353LL;
const int maxn = 3000010;
ll qpow(ll a,ll x){
    ll ret=1;
    while (x){
        if (x&1)
            ret = ret*a%mod;
        a=a*a%mod;
        x>>=1;
    }
    return ret;
}
ll fac[maxn],inv[maxn];
 
ll init(){
    fac[0]=1;
    for (int i=1;i<maxn;i++)
        fac[i]=fac[i-1]*i%mod;
    inv[maxn-1]=qpow(fac[maxn-1],mod-2);
    for (int i=maxn-2;i>=0;i--)
        inv[i]=inv[i+1]*(i+1)%mod;
    return 0;
}
 
ll c(ll n,ll m){
    if (n<m) return 0;
    return fac[n]*inv[m]%mod*inv[n-m]%mod;
}
int main()
{
    init();
    ll N,A,B,K;
    cin>>N>>A>>B>>K;
    ll ans=0;
    for(int i=0;i<=N;i++){
        if(K-A*i<0) continue;
        if((K-A*i)%B!=0) continue;
        ll y=(K-A*i)/B;
        if(y>N)continue;
        ans=(ans+c(N,i)*c(N,y)+mod)%mod;
    }
    cout<<ans<<endl;
    return 0;
}