Submission #65673700

---

Source Code Expand

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

const int MOD=998244353;

vector<long long> nfact, nfactInv;

long long modpow(long long x, long long n, long long m) {
    if (n == 0) return 1%m;
    long long u = modpow(x,n/2,m);
    u = (u*u)%m;
    if (n%2 == 1) u = (u*x)%m;
    return u;
}
 
long long ncr(long long n, long long r,int N=4000000){
	if(!nfact.size()){
	    nfact.resize(N+1);
	    nfact[0]=1;
		for(int i=1; i<=N; i++){
			nfact[i]=nfact[i-1]*i;
			nfact[i]%=MOD;
		}
		
		nfactInv.resize(N+1);
		nfactInv[N]=modpow(nfact[N],MOD-2,MOD);
		for(int i=N-1; i>=0; i--){
		    nfactInv[i]=nfactInv[i+1]*(i+1);
			nfactInv[i]%=MOD;
		}
	}
	
	
    if(r>n || r<0) return 0;
    long long d1=nfactInv[r];
    long long d2=nfactInv[n-r];
    long long mult=(d1*d2)%MOD;
    return (nfact[n]*mult)%MOD;
}


int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
	int A,O,B,G;
	cin>>A>>O>>B>>G;
	
	int tot=A+O+B+G;
	long long ans=0;
	
	for(int i=A;i<=A+O;i++){
		
		long long val=ncr(i-1,A-1)*ncr(tot-i,G+O-i+A);
		
		
		val%=MOD;
		ans+=val;
		ans%=MOD;
		
		
		
	}
	cout<<ans; 
    
    return 0;
}

Submission Info

Judge Result