#774720 (C++14) No.1963 Subset Mex

• 問題: No.1963 Subset Mex
• ユーザー: Suiseiseki
• 提出日時: 2022-07-01 18:45:31
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 204 ms / 4,000 ms
• コード長: 1,172 bytes
• コンパイル時間: 476 ms
• コンパイル使用メモリ: 43,300 KB
• 実行使用メモリ: 7,332 KB
• 最終ジャッジ日時: 2023-08-17 05:54:49
• 合計ジャッジ時間: 3,475 ms
• ジャッジサーバーID （参考情報）: judge12 / judge14

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 4,380 KB
testcase_01	AC	2 ms; 4,376 KB
testcase_02	AC	2 ms; 4,376 KB
testcase_03	AC	2 ms; 4,376 KB
testcase_04	AC	3 ms; 4,376 KB
testcase_05	AC	3 ms; 4,376 KB
testcase_06	AC	4 ms; 4,376 KB
testcase_07	AC	3 ms; 4,380 KB
testcase_08	AC	3 ms; 4,376 KB
testcase_09	AC	2 ms; 4,376 KB
testcase_10	AC	2 ms; 4,376 KB
testcase_11	AC	3 ms; 4,376 KB
testcase_12	AC	10 ms; 7,152 KB
testcase_13	AC	12 ms; 7,288 KB
testcase_14	AC	14 ms; 7,292 KB
testcase_15	AC	14 ms; 7,228 KB
testcase_16	AC	16 ms; 7,240 KB
testcase_17	AC	18 ms; 7,328 KB
testcase_18	AC	25 ms; 7,256 KB
testcase_19	AC	54 ms; 7,324 KB
testcase_20	AC	87 ms; 7,220 KB
testcase_21	AC	175 ms; 7,280 KB
testcase_22	AC	204 ms; 7,280 KB
testcase_23	AC	168 ms; 7,332 KB
testcase_24	AC	124 ms; 7,256 KB
testcase_25	AC	103 ms; 7,292 KB
testcase_26	AC	18 ms; 7,300 KB
testcase_27	AC	26 ms; 7,320 KB
testcase_28	AC	49 ms; 7,280 KB
testcase_29	AC	61 ms; 7,288 KB

ソースコード

#include <cstdio>
#include <algorithm>
const int Maxn=100;
const int Mod=998244353;
int n;
int a[Maxn+5];
int f[Maxn+5][Maxn+5][Maxn+5];
int find_last(int n){
	static int g[Maxn+5];
	g[n]=1;
	int ans=0;
	for(int i=n;i>0;i--){
		int use=std::min(g[i],a[i]);
		g[i-1]=std::min(g[i]*2-use,10000);
		ans-=a[i]-use;
	}
	ans+=g[0]-a[0];
	return ans;
}
int main(){
	scanf("%d",&n);
	int mn_val=Maxn;
	for(int i=1;i<=n;i++){
		int x;
		scanf("%d",&x);
		mn_val=std::min(mn_val,x);
		a[x]++;
	}
	f[Maxn+1][0][0]=1;
	int ans=0;
	for(int i=Maxn;i>=0;i--){
		int require=find_last(i-1);
		for(int j=0;j<=n;j++){
			for(int k=0;k<=n;k++){
				if(f[i+1][j][k]==0){
					continue;
				}
				for(int t=0;t<=n;t++){
					int use=std::min(j+t,a[i]);
					int nxt_j=2*j+t-use,nxt_k=k+a[i]-use;
					if(i==0&&nxt_j>0&&j+t<=a[i]+k){
						ans=(ans+f[i+1][j][k])%Mod;
					}
					if(nxt_j<0||nxt_j>n||nxt_k<0||nxt_k>n){
						continue;
					}
					if((nxt_k!=0||mn_val!=i)&&nxt_j==0&&nxt_k+1>=require&&t>0){
						ans=(ans+f[i+1][j][k])%Mod;
					}
					if(i>0){
						f[i][nxt_j][nxt_k]=(f[i][nxt_j][nxt_k]+f[i+1][j][k])%Mod;
					}
				}
			}
		}
	}
	printf("%d\n",(ans+1)%Mod);
	return 0;
}