結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 31 ms
118,712 KB
testcase_01 AC 31 ms
118,688 KB
testcase_02 AC 29 ms
118,712 KB
testcase_03 AC 30 ms
118,820 KB
testcase_04 AC 31 ms
118,736 KB
testcase_05 AC 30 ms
118,748 KB
testcase_06 AC 31 ms
118,800 KB
testcase_07 AC 30 ms
118,748 KB
testcase_08 AC 31 ms
118,692 KB
testcase_09 AC 30 ms
118,832 KB
testcase_10 AC 29 ms
118,752 KB
testcase_11 AC 65 ms
118,684 KB
testcase_12 AC 214 ms
118,776 KB
testcase_13 AC 129 ms
118,916 KB
testcase_14 AC 104 ms
118,768 KB
testcase_15 AC 195 ms
118,680 KB
testcase_16 AC 72 ms
118,688 KB
testcase_17 AC 166 ms
118,712 KB
testcase_18 AC 181 ms
118,748 KB
testcase_19 AC 95 ms
118,720 KB
testcase_20 AC 152 ms
118,796 KB
testcase_21 AC 134 ms
118,788 KB
testcase_22 AC 111 ms
118,720 KB
testcase_23 AC 154 ms
118,680 KB
testcase_24 AC 148 ms
118,872 KB
testcase_25 AC 200 ms
118,744 KB
testcase_26 AC 140 ms
118,764 KB
testcase_27 AC 245 ms
118,860 KB
testcase_28 AC 229 ms
118,788 KB
testcase_29 AC 241 ms
118,896 KB
testcase_30 AC 239 ms
118,708 KB
testcase_31 AC 230 ms
118,688 KB
testcase_32 AC 235 ms
118,716 KB
testcase_33 AC 245 ms
118,704 KB
testcase_34 AC 248 ms
118,744 KB
testcase_35 AC 242 ms
118,772 KB
testcase_36 AC 30 ms
118,788 KB
testcase_37 AC 31 ms
118,764 KB
testcase_38 AC 30 ms
118,752 KB
testcase_39 AC 33 ms
118,760 KB
testcase_40 AC 29 ms
118,848 KB
testcase_41 AC 32 ms
118,724 KB
testcase_42 AC 70 ms
118,756 KB
testcase_43 AC 229 ms
118,760 KB
testcase_44 AC 32 ms
118,824 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)+2;

	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