結果

問題 No.1646 Avoid Palindrome
ユーザー monnu
提出日時 2021-08-29 20:26:46
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 422 ms / 3,000 ms
コード長 1,880 bytes
コンパイル時間 3,947 ms
コンパイル使用メモリ 231,912 KB
実行使用メモリ 269,440 KB
最終ジャッジ日時 2024-11-22 08:14:49
合計ジャッジ時間 17,235 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using ll=long long;
#define MOD 998244353
using Graph=vector<vector<int>>;
#define INF 1000000000
#define MAX 200000

int main(){
  int N;
  string S;
  cin>>N>>S;
  if(N==1){
    if(S=="?"){
      cout<<26<<'\n';
    }else{
      cout<<1<<'\n';
    }
    return 0;
  }

  vector<vector<ll>> dp(N,vector<ll>(26*26,0));
  if(S[0]=='?'&&S[1]=='?'){
    for(int a=0;a<26;a++){
      for(int b=0;b<26;b++){
        if(a==b){
          continue;
        }
        dp[1][26*a+b]=1;
      }
    }
  }else if(S[0]=='?'){
    for(int a=0;a<26;a++){
      if(a==S[1]-'a'){
        continue;
      }
      dp[1][26*a+S[1]-'a']=1;
    }
  }else if(S[1]=='?'){
    for(int b=0;b<26;b++){
      if(S[0]-'a'==b){
        continue;
      }
      dp[1][26*(S[0]-'a')+b]=1;
    }
  }else{
    if(S[0]!=S[1]){
      dp[1][26*(S[0]-'a')+S[1]-'a']=1;
    }
  }
  for(int i=2;i<N;i++){
    if(S[i]!='?'){
      for(int a=0;a<26;a++){
        if(a==S[i]-'a'){
          continue;
        }
        for(int b=0;b<26;b++){
          if(b==S[i]-'a'){
            continue;
          }
          if(a==b){
            continue;
          }
          dp[i][26*b+(S[i]-'a')]+=dp[i-1][26*a+b];
          dp[i][26*b+(S[i]-'a')]%=MOD;
        }
      }
    }else{
      for(int b=0;b<26;b++){
        ll sum=0;
        for(int a=0;a<26;a++){
          sum+=dp[i-1][26*a+b];
          sum%=MOD;
        }
        for(int c=0;c<26;c++){
          if(b==c){
            continue;
          }
          dp[i][26*b+c]=sum;
        }
        for(int a=0;a<26;a++){
          if(a==b){
            continue;
          }
          dp[i][26*b+a]+=MOD-dp[i-1][26*a+b];
          dp[i][26*b+a]%=MOD;
        }
      }
    }
  }

  ll ans=0;
  for(int j=0;j<26*26;j++){
    ans+=dp[N-1][j];
    ans%=MOD;
  }
  cout<<ans<<'\n';
}
0