結果

問題 No.1646 Avoid Palindrome
コンテスト
ユーザー manjuuu_onsen
提出日時 2021-08-13 22:51:46
言語 C++14
(gcc 15.2.0 + boost 1.90.0)
コンパイル:
g++-15 -O2 -lm -std=c++14 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 962 ms / 3,000 ms
コード長 1,581 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,371 ms
コンパイル使用メモリ 212,340 KB
実行使用メモリ 178,048 KB
最終ジャッジ日時 2026-06-25 13:26:15
合計ジャッジ時間 21,220 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#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;

	if(N == 1) {
		if(s[0] == '?') cout << 26 << endl;
		else cout << 1 << endl;
		return 0;
	}

	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