Submission #60166594

---

Source Code Expand

#include<bits/stdc++.h>
#include"atcoder/all"
using namespace std;
using namespace atcoder;
#define rep(i,n) for(int i=0;i<(n);i++)
#define all(a) a.begin(),a.end()
typedef long long ll;
typedef vector<ll> vi;
typedef pair<ll,ll> P;
const ll mod=998244353;
const ll inf=1ll<<61;
typedef modint998244353 mi;

mi kj[600005];

mi binom(int r,int c){
	if(r<c)return 0;
	return kj[r]/kj[c]/kj[r-c];
}

int main(){
	kj[0]=1;
	for(int i=1;i<=600000;i++)kj[i]=kj[i-1]*i;

	int R,G,B,K,X,Y,Z;
	cin>>R>>G>>B>>K>>X>>Y>>Z;

	vector<mi>r(R+1),b(B+1);
	rep(i,R+1){
		if(K-Y<=i)r[i]=binom(R,i);
		else r[i]=0;
	}
	rep(i,B+1){
		if(K-X<=i)b[i]=binom(B,i);
		else b[i]=0;
	}
	vector<mi>rb=convolution(r,b);
	for(int i=Z+1;i<rb.size();i++)rb[i]=0;
	vector<mi>g(G+1);
	rep(i,G+1){
		g[i]=binom(G,i);
	}
	vector<mi>rgb=convolution(rb,g);
	cout<<rgb[K].val()<<endl;
}

Submission Info

Compile Error

Main.cpp: In function ‘int main()’:
Main.cpp:38:24: warning: comparison of integer expressions of different signedness: ‘int’ and ‘std::vector<atcoder::static_modint<998244353> >::size_type’ {aka ‘long unsigned int’} [-Wsign-compare]
   38 |         for(int i=Z+1;i<rb.size();i++)rb[i]=0;
      |                       ~^~~~~~~~~~

Judge Result