結果

問題 No.1646 Avoid Palindrome
コンテスト
ユーザー manjuuu_onsen
提出日時 2021-08-13 22:50:21
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 1,486 bytes
コンパイル時間 2,650 ms
コンパイル使用メモリ 195,080 KB
実行使用メモリ 178,048 KB
最終ジャッジ日時 2024-10-03 21:26:25
合計ジャッジ時間 30,219 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 38 RE * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include<bits/stdc++.h>

using namespace std;
using ll = long long;
using pint = pair<int,int>;

#include"atcoder/modint"
using mint = atcoder::modint998244353; 
ostream &operator<<(std::ostream &os,const mint &p){return os << p.val();}
istream &operator>>(std::istream &is, mint &a){int64_t x;is >> x;a = mint(x);return (is);}


int main()
{
	int N; cin >> N;
	string s; cin >> s;
	constexpr int AL = 26;
	for(auto &i : s) {
		i = (i == '?' ? AL : i - 'a');
	}

	vector<vector<vector<mint>>> dp(N + 1, vector<vector<mint>>(AL, vector<mint>(AL)));
	for(int i = 0; i < AL; i++) {
		for(int j = 0; j < AL; j++) {
			dp[2][i][j] = (s[0] == i || s[0] == AL) && (s[1] == j || s[1] == AL);
			if(i == j)dp[2][i][j] = 0;
		}
	}

	for(int i = 2; i < N; i++) {
		for(int f = 0; f < AL; f++) {
			if(s[i - 2] != AL && s[i - 2] != f)continue;
			for(int p = 0; p < AL; p++) {
				if(s[i - 1] != AL && s[i - 1] != p)continue;
				if(s[i] != AL){
					if(f != s[i] && p != s[i])dp[i + 1][p][s[i]] += dp[i][f][p];
				}
				else {
					for(int j = 0; j < AL; j++) {
						
						if(f != j && p != j) {
							dp[i + 1][p][j] += dp[i][f][p];
						}
					}
				} 
			}
		}

	}


	mint sum = 0;
	for(int i = 0; i < AL; i++) {
		if(s[N - 2] != AL && s[N - 2] != i) continue;
		for(int j = 0; j < AL; j++) {
			if(s[N - 1] != AL && s[N - 1] != j) continue;

			sum += dp[N][i][j];
		}
	}	
	cout << sum << endl;

}
0