結果

問題 No.1963 Subset Mex
ユーザー 1kri1kri
提出日時 2022-07-14 10:29:03
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 307 ms / 4,000 ms
コード長 1,361 bytes
コンパイル時間 607 ms
コンパイル使用メモリ 64,896 KB
実行使用メモリ 13,568 KB
最終ジャッジ日時 2024-06-25 17:31:26
合計ジャッジ時間 3,733 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 8 ms
13,440 KB
testcase_01 AC 8 ms
13,440 KB
testcase_02 AC 9 ms
13,440 KB
testcase_03 AC 9 ms
13,312 KB
testcase_04 AC 8 ms
13,440 KB
testcase_05 AC 9 ms
13,440 KB
testcase_06 AC 9 ms
13,568 KB
testcase_07 AC 9 ms
13,440 KB
testcase_08 AC 9 ms
13,440 KB
testcase_09 AC 9 ms
13,312 KB
testcase_10 AC 9 ms
13,568 KB
testcase_11 AC 9 ms
13,440 KB
testcase_12 AC 14 ms
13,440 KB
testcase_13 AC 15 ms
13,440 KB
testcase_14 AC 17 ms
13,440 KB
testcase_15 AC 19 ms
13,440 KB
testcase_16 AC 20 ms
13,440 KB
testcase_17 AC 22 ms
13,312 KB
testcase_18 AC 33 ms
13,312 KB
testcase_19 AC 77 ms
13,440 KB
testcase_20 AC 128 ms
13,440 KB
testcase_21 AC 264 ms
13,440 KB
testcase_22 AC 307 ms
13,440 KB
testcase_23 AC 251 ms
13,440 KB
testcase_24 AC 184 ms
13,440 KB
testcase_25 AC 152 ms
13,440 KB
testcase_26 AC 21 ms
13,312 KB
testcase_27 AC 33 ms
13,440 KB
testcase_28 AC 70 ms
13,440 KB
testcase_29 AC 89 ms
13,440 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <cstdio>
#include <cstring>
#define mod 998244353
using namespace std;
int t,n,m,x,cnt[115],sum[115],f[115][105][105],ok[115][105][105];
void upd(int &x,int y){
	x+=y;
	if (x>=mod)x-=mod;
	return;
}
int check(int now,int rest,int d){
	if (d>n)return 0;
	if (now==0){
		if (d<=rest+cnt[now])return 1;
		return 0;
	}
	if (ok[now][rest][d]!=-1)return ok[now][rest][d];
	if (cnt[now]>=d)return ok[now][rest][d]=check(now-1,rest+cnt[now]-d,d);
	return ok[now][rest][d]=check(now-1,rest,d+(d-cnt[now]));
}
int dfs(int now,int rest,int d){
	if (d>n)return 0;
	if (f[now][rest][d]!=-1)return f[now][rest][d];
	int ans=0;
	for (int i=0;i<=n;i++){
		int _rest=rest,_d=d;
		if (cnt[now]>=i+d)_rest+=cnt[now]-(i+d);
		else _d+=(i+d)-cnt[now];
		if (now>0)upd(ans,dfs(now-1,_rest,_d));
		if (i>0){
			if (now==0){
				if (i+d<=rest+cnt[now])upd(ans,1);
			}
			else{
				if (_d>0){
					if (check(now-1,_rest,_d))upd(ans,1);
				}
				else if ((_rest==0&&sum[now-1]==0)||check(now-1,_rest+1,1))upd(ans,1);
			}
		}
	}
	return f[now][rest][d]=ans;
}
int main(){
    t=1;
	while(t--){
		cin>>n;
		memset(cnt,0,sizeof(cnt));
		for (int i=1;i<=n;i++){
			cin>>x;
			cnt[x]++;
		}
		m=114;
		sum[0]=cnt[0];
		for (int i=1;i<=m;i++)sum[i]=cnt[i]+sum[i-1];
		memset(f,-1,sizeof(f));
		memset(ok,-1,sizeof(ok));
		cout<<dfs(m,0,0)<<endl;
	}
	return 0;
}
0