#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
• 実行時間: 217 ms / 4,000 ms
• コード長: 1,172 bytes
• コンパイル時間: 309 ms
• コンパイル使用メモリ: 43,844 KB
• 実行使用メモリ: 7,552 KB
• 最終ジャッジ日時: 2024-05-04 12:50:01
• 合計ジャッジ時間: 2,633 ms
• ジャッジサーバーID （参考情報）: judge5 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	3 ms; 5,376 KB
testcase_05	AC	3 ms; 5,376 KB
testcase_06	AC	3 ms; 5,376 KB
testcase_07	AC	3 ms; 5,376 KB
testcase_08	AC	3 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	2 ms; 5,376 KB
testcase_11	AC	2 ms; 5,376 KB
testcase_12	AC	10 ms; 7,424 KB
testcase_13	AC	11 ms; 7,552 KB
testcase_14	AC	14 ms; 7,424 KB
testcase_15	AC	13 ms; 7,296 KB
testcase_16	AC	15 ms; 7,296 KB
testcase_17	AC	18 ms; 7,424 KB
testcase_18	AC	25 ms; 7,424 KB
testcase_19	AC	60 ms; 7,424 KB
testcase_20	AC	99 ms; 7,424 KB
testcase_21	AC	186 ms; 7,424 KB
testcase_22	AC	217 ms; 7,552 KB
testcase_23	AC	179 ms; 7,424 KB
testcase_24	AC	133 ms; 7,552 KB
testcase_25	AC	106 ms; 7,552 KB
testcase_26	AC	17 ms; 7,424 KB
testcase_27	AC	27 ms; 7,424 KB
testcase_28	AC	52 ms; 7,424 KB
testcase_29	AC	65 ms; 7,552 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;
}