Atcoder 700 RGB Coloring 组合数

题意:n件物品,初始都为白色,现在将物品涂色,涂红色+A分,涂蓝色+B分,涂绿色加(A+B)分.
1<=n,a,b<=3e5. 0<=k<=18e10.可以有物品不涂色,问n件物品的得分正好为K的方案数mod998244353?


若a>b 则swap(a,b)
一件物品的得分为0,a,b,a+b. 枚举物品得分至少为a的有x件,那么还要在选m=(K-A*x)/B 件物品得分+B
物品得分至少为a的有x件,方案为f[x]=C(n,x) * C(n,m) 

正好有x件物品得分为a的有f[x]-f[x+1].

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=3e5+5,mod=998244353;
ll n,a,b,k,fac[N],f[N];
void init(){
	fac[0]=1;
	for(ll i=1;i<N;i++)	fac[i]=(fac[i-1]*i)%mod;
}
ll powmod(ll x,ll n){
	ll s=1;
	while(n){
		if(n&1)	s=(s*x)%mod;
		x=(x*x)%mod;
		n>>=1;	
	}
	return s;
}
ll C(int n,int m){
	if(n<m)
		return 0;
	ll a=fac[n],b=(fac[n-m]*fac[m])%mod;
	return (a*powmod(b,mod-2))%mod;
}
int main()
{
	init();
	cin>>n>>a>>b>>k;
	if(a>b)	swap(a,b);
	ll res=0;
	for(ll x=0;x<=n;x++)
	{
		ll ned=k-x*a;
		if(ned<0||ned%b)	continue;
		ll m=ned/b;
		f[x]=((C(n,x)*C(n,m))%mod - f[x+1] + mod )%mod;
		res=(res+f[x])%mod;
	}
	cout<<res<<'\n';
	return 0;
}