結果

問題 No.1646 Avoid Palindrome
コンテスト
ユーザー 夜
提出日時 2021-08-14 09:42:36
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 746 ms / 3,000 ms
コード長 1,607 bytes
コンパイル時間 991 ms
コンパイル使用メモリ 94,064 KB
実行使用メモリ 355,328 KB
最終ジャッジ日時 2024-11-08 16:13:09
合計ジャッジ時間 24,961 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
#include <iomanip>
#include <cmath>
#include <stdio.h>
#include <queue>
#include <deque>
#include <cstdio>
#include <set>
#include <map>
#include <bitset>
#include <stack>
#include <cctype>
using namespace std;
long long mod = 998244353;
long long dp[50050][30][30] = {};
long long sum1[30] = {};
int main() {
	long long n, co = 0;
	string s;
	cin >> n >> s;
	if (n == 1) {
		if (s == "?") {
			cout << 26 << endl;
		}
		else {
			cout << 1 << endl;
		}
		return 0;
	}
	for (int i = 0; i < 26; i++) {
		for (int j = 0; j < 26; j++) {
			int i1 = i, j1 = j;
			if (s[0] != '?') {
				i1 = s[0] - 'a';
			}
			if (s[1] != '?') {
				j1 = s[1] - 'a';
			}
			if (i1 != j1) {
				dp[2][j1][i1] = 1;
			}
		}
	}
	for (int j = 0; j < 26; j++) {
		for (int k = 0; k < 26; k++) {
			sum1[j] += dp[2][j][k];
		}
	}
	for (int i = 2; i < n; i++) {
		if (s[i] == '?') {
			for (int j = 0; j < 26; j++) {
				for (int k = 0; k < 26; k++) {
					if (j != k) {
						dp[i + 1][j][k] += sum1[k];
						dp[i + 1][j][k] -= dp[i][k][j];
					}
				}
			}
		}
		else {
			for (int j = 0; j < 26; j++) {
				if (s[i] - 'a' != j) {
					dp[i + 1][s[i] - 'a'][j] += sum1[j];
					dp[i + 1][s[i] - 'a'][j] -= dp[i][j][s[i] - 'a'];
				}
			}
		}
		for (int j = 0; j < 26; j++) {
			sum1[j] = 0;
			for (int k = 0; k < 26; k++) {
				dp[i + 1][j][k] %= mod;
				sum1[j] += dp[i + 1][j][k];
			}
		}
	}
	long long ans = 0;
	for (int i = 0; i < 26; i++) {
		for (int j = 0; j < 26; j++) {
			ans += dp[n][i][j];
			ans %= mod;
		}
	}
	cout << ans << endl;
}
0