結果

問題 No.2951 Similar to Mex
ユーザー nouka28nouka28
提出日時 2024-09-14 12:06:14
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 254 ms / 2,000 ms
コード長 1,603 bytes
コンパイル時間 5,395 ms
コンパイル使用メモリ 309,248 KB
実行使用メモリ 118,968 KB
最終ジャッジ日時 2024-10-25 20:51:05
合計ジャッジ時間 11,915 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 32 ms
118,832 KB
testcase_01 AC 32 ms
118,748 KB
testcase_02 AC 31 ms
118,760 KB
testcase_03 AC 33 ms
118,848 KB
testcase_04 AC 32 ms
118,784 KB
testcase_05 AC 33 ms
118,700 KB
testcase_06 AC 32 ms
118,696 KB
testcase_07 AC 32 ms
118,768 KB
testcase_08 AC 33 ms
118,680 KB
testcase_09 AC 32 ms
118,932 KB
testcase_10 AC 32 ms
118,856 KB
testcase_11 AC 68 ms
118,752 KB
testcase_12 AC 224 ms
118,784 KB
testcase_13 AC 134 ms
118,744 KB
testcase_14 AC 110 ms
118,672 KB
testcase_15 AC 201 ms
118,968 KB
testcase_16 AC 77 ms
118,768 KB
testcase_17 AC 177 ms
118,776 KB
testcase_18 AC 192 ms
118,772 KB
testcase_19 AC 97 ms
118,768 KB
testcase_20 AC 161 ms
118,816 KB
testcase_21 AC 142 ms
118,716 KB
testcase_22 AC 113 ms
118,700 KB
testcase_23 AC 163 ms
118,908 KB
testcase_24 AC 154 ms
118,880 KB
testcase_25 AC 210 ms
118,696 KB
testcase_26 AC 140 ms
118,744 KB
testcase_27 AC 251 ms
118,692 KB
testcase_28 AC 236 ms
118,688 KB
testcase_29 AC 254 ms
118,868 KB
testcase_30 AC 245 ms
118,744 KB
testcase_31 AC 240 ms
118,780 KB
testcase_32 AC 239 ms
118,704 KB
testcase_33 AC 245 ms
118,772 KB
testcase_34 AC 251 ms
118,748 KB
testcase_35 AC 248 ms
118,736 KB
testcase_36 AC 32 ms
118,768 KB
testcase_37 AC 33 ms
118,900 KB
testcase_38 AC 32 ms
118,728 KB
testcase_39 AC 34 ms
118,748 KB
testcase_40 AC 32 ms
118,912 KB
testcase_41 AC 34 ms
118,852 KB
testcase_42 AC 74 ms
118,856 KB
testcase_43 AC 237 ms
118,772 KB
testcase_44 AC 35 ms
118,744 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#include<atcoder/all>
using namespace atcoder;
using mint=atcoder::modint998244353;

template<long long mod,long long MAX_N>
struct factional_prime{
	long long inv_[MAX_N+1];
	long long fac_[MAX_N+1];
	long long fac_inv_[MAX_N+1];

	factional_prime(){
		inv_[0]=0;inv_[1]=fac_[0]=fac_[1]=fac_inv_[0]=fac_inv_[1]=1;
		for(long long i=2;i<=MAX_N;i++){
			inv_[i]=((mod-mod/i)*inv_[mod%i])%mod;
			fac_[i]=(fac_[i-1]*i)%mod;
			fac_inv_[i]=(fac_inv_[i-1]*inv_[i])%mod;
		}
	}
	long long inv(long long n){
		if(n<0)return 0;
		return inv_[n];
	}
	long long fac(long long n){
		if(n<0)return 0;
		return fac_[n];
	}
	long long finv(long long n){
		if(n<0)return 0;
		return fac_inv_[n];
	}
	long long nCr(long long n,long long r){
		if(n<r||n<0||r<0)return 0;
		return ((fac_[n]*fac_inv_[n-r])%mod*fac_inv_[r])%mod;
	}
	long long nPr(long long n,long long r){
		if(n<r||n<0||r<0)return 0;
		return (fac_[n]*fac_inv_[n-r])%mod;
	}
};

factional_prime<998244353,1000> fp;

mint dp[309][309][309];

int main(){
	int N,M,K;cin>>N>>M>>K;

	dp[0][0][0]=1;

	int T=max(M,K)+1;

	for(int i=1;i<=T;i++){
		for(int j=0;j<=min(i-1,N);j++){
			for(int k=0;k<=i-1;k++){
				//0
				dp[i][j][i]+=dp[i-1][j][k]*mint(i).pow(max(0,min(K,i)-k));
				
				if(i<=M&&j<N){
					//1
					dp[i][j+1][k]+=dp[i-1][j][k];
				}
			}
		}
	}

	mint ans=0;

	for(int i=1;i<=N;i++){
		mint sum=0;
		for(int j=0;j<=i;j++){
			sum+=mint(j).pow(N)*mint(-1).pow(i-j)*fp.nCr(i,j);
		}
		mint sum2=0;
		for(int j=0;j<=T;j++){
			sum2+=dp[T][i][j];
		}
		ans+=sum*sum2;
	}

	cout<<ans.val()<<endl;
}
0