結果

問題 No.1646 Avoid Palindrome
ユーザー SSRS
提出日時 2021-08-13 21:28:19
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2,638 ms / 3,000 ms
コード長 1,259 bytes
コンパイル時間 1,849 ms
コンパイル使用メモリ 174,224 KB
実行使用メモリ 320,256 KB
最終ジャッジ日時 2024-11-08 14:54:20
合計ジャッジ時間 73,478 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
const long long MOD = 998244353;
int main(){
  int N;
  cin >> N;
  string S;
  cin >> S;
  if (N == 1){
    if (S[0] == '?'){
      cout << 26 << endl;
    } else {
      cout << 1 << endl;
    }
  } else {
    vector<vector<vector<long long>>> dp(N + 1, vector<vector<long long>>(26, vector<long long>(26, 0)));
    for (int i = 0; i < 26; i++){
      if (S[0] == '?' || S[0] - 'a' == i){
        for (int j = 0; j < 26; j++){
          if (S[1] == '?' || S[1] - 'a' == j){
            if (i != j){
              dp[2][i][j]++;
            }
          }
        }
      }
    }
    for (int i = 2; i < N; i++){
      for (int j = 0; j < 26; j++){
        for (int k = 0; k < 26; k++){
          for (int l = 0; l < 26; l++){
            if (S[i] == '?' || S[i] - 'a' == l){
              if (j != l && k != l){
                dp[i + 1][k][l] += dp[i][j][k];
                if (dp[i + 1][k][l] >= MOD){
                  dp[i + 1][k][l] -= MOD;
                }
              }
            }
          }
        }
      }
    }
    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