提出 #41154363

---

ソースコード 拡げる

#include <cstdio>
#include <vector>
#include <cstring>
#include <algorithm>
using namespace std;
#ifdef LOCAL
#define debug(...) fprintf(stderr,##__VA_ARGS__)
#else
#define debug(...) void(0)
#endif
typedef long long LL;
const int P=998244353;
LL mod(LL x){return (x%P+P)%P;}
void red(LL&x){x=mod(x);}
LL qpow(LL a,LL b,int p=P){LL r=1;for(a%=p;b;b>>=1,a=a*a%p) if(b&1) r=r*a%p; return r;}
template<int N,int P> struct C_prime{
	LL fac[N+10],ifac[N+10];
	C_prime(){
		for(int i=fac[0]=ifac[0]=1;i<=N;i++) fac[i]=fac[i-1]*i%P;
		ifac[N]=qpow(fac[N],P-2,P);
		for(int i=N-1;i>=1;i--) ifac[i]=ifac[i+1]*(i+1)%P;
	}
	LL operator()(int n,int m){return n>=m?fac[n]*ifac[n-m]%P*ifac[m]%P:0;}
};
int n,m,k,X,buc[1<<11],sum[1<<11];
C_prime<1<<20,P> Z;
LL f[1<<12][1<<11],g[1<<12][1<<11];
LL dp(){
	f[0][0]=0,g[0][0]=1;
	for(int j=1;j<=m;j++){
		auto calc=[&](int l,int r)->int{
			auto len=[](int l,int r)->int{return max(r-l+1,0);};
			return len(l,r)-len(max(l,X),min(r,X+k-1));
		};
		for(int i=sum[j-1];i<=n+k;i++){
			for(int x=buc[j];i+x<=n+k;x++){
				LL z=Z(k-(i-sum[j-1]),x-buc[j]);
				red(g[i+x][j]+=g[i][j-1]*z);
				red(f[i+x][j]+=(f[i][j-1]+g[i][j-1]*calc(i+1,i+x)*j%P)*z);
			}
		}
#ifdef LOCAL
		for(int i=sum[j];i<=n+k;i++) debug("i=%d,j=%d,f=%lld,g=%lld\n",i,j,f[i][j],g[i][j]);
#endif
	}
	return f[n+k][m];
}
int main(){
//	#ifdef LOCAL
//	 	freopen("input.in","r",stdin);
//	#endif
	scanf("%d%d%d%d",&n,&m,&k,&X);
	for(int i=1,x;i<=n;i++) scanf("%d",&x),buc[x]++;
	for(int i=1;i<=m;i++) sum[i]=sum[i-1]+buc[i];
	printf("%lld\n",dp());
	return 0;
}

提出情報

コンパイルエラー

./Main.cpp: In function ‘int main()’:
./Main.cpp:53:7: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   53 |  scanf("%d%d%d%d",&n,&m,&k,&X);
      |  ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
./Main.cpp:54:31: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   54 |  for(int i=1,x;i<=n;i++) scanf("%d",&x),buc[x]++;
      |                          ~~~~~^~~~~~~~~

ジャッジ結果