#1012140 (C++17) No.2899 Taffy Permutation

• 問題: No.2899 Taffy Permutation
• ユーザー: 沙耶花
• 提出日時: 2024-09-20 22:06:21
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 14 ms / 2,000 ms
• コード長: 1,079 bytes
• コンパイル時間: 4,144 ms
• コンパイル使用メモリ: 268,488 KB
• 実行使用メモリ: 6,944 KB
• 最終ジャッジ日時: 2024-09-20 22:06:27
• 合計ジャッジ時間: 4,917 ms
• ジャッジサーバーID （参考情報）: judge3 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 6,812 KB
testcase_01	AC	1 ms; 6,816 KB
testcase_02	AC	1 ms; 6,820 KB
testcase_03	AC	1 ms; 6,940 KB
testcase_04	AC	2 ms; 6,944 KB
testcase_05	AC	1 ms; 6,944 KB
testcase_06	AC	2 ms; 6,940 KB
testcase_07	AC	1 ms; 6,944 KB
testcase_08	AC	1 ms; 6,944 KB
testcase_09	AC	14 ms; 6,940 KB
testcase_10	AC	13 ms; 6,944 KB
testcase_11	AC	14 ms; 6,940 KB
testcase_12	AC	7 ms; 6,940 KB
testcase_13	AC	8 ms; 6,940 KB
testcase_14	AC	8 ms; 6,944 KB
testcase_15	AC	11 ms; 6,940 KB
testcase_16	AC	12 ms; 6,940 KB
testcase_17	AC	11 ms; 6,944 KB
testcase_18	AC	2 ms; 6,940 KB
testcase_19	AC	11 ms; 6,940 KB

ソースコード

#include <stdio.h>
#include <atcoder/all>
#include <bits/stdc++.h>
using namespace std;
using namespace atcoder;
using mint = modint998244353;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf32 1000000001
#define Inf64 4000000000000000001LL

int main(){
	
	int n;
	cin>>n;
	string s;
	cin>>s;
	vector<mint> dp;
	int pos = -1;
	rep(i,n){
		if(s[i]=='0'){
			pos = i;break;
		}
	}
	if(pos==-1){
		mint ans = 1;
		rep(i,n)ans *= i+1;
		cout<<ans.val()<<endl;
		return 0;
	}
	{
		mint t = 1;
		rep(i,pos)t *= i+1;
		dp.resize(pos+1,t);
	}
	for(int i = pos+1;i<n;i++){
		rep(j,dp.size()){
		//	cout<<dp[j].val()<<' ';
		}
		//cout<<endl;
		if(s[i]=='1'){
			vector<mint> ndp(dp.size());
			rep(j,dp.size()){
				ndp[j] = dp[j] * (i-j);
			}
			dp = ndp;
			continue;
		}
		vector<mint> ndp(i+1);
		while(dp.size()!=ndp.size())dp.push_back(0);
		rep(j,dp.size()-1)ndp[j+1] += dp[j] * (j+1);
		rep(j,dp.size()-1){
			dp[j+1] += dp[j];
		}
		rep(j,dp.size()-1)ndp[j+1] += dp[j];
		dp = ndp;
	}
	mint ans = 0;
	rep(i,dp.size())ans += dp[i];
	cout<<ans.val()<<endl;
	return 0;
}