結果

問題 No.1646 Avoid Palindrome
ユーザー yunix91201367yunix91201367
提出日時 2021-08-13 22:20:08
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 616 ms / 3,000 ms
コード長 2,986 bytes
コンパイル時間 2,148 ms
コンパイル使用メモリ 179,980 KB
実行使用メモリ 320,128 KB
最終ジャッジ日時 2024-11-08 15:30:42
合計ジャッジ時間 20,903 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/math>
using namespace std;
using namespace atcoder;
#define int long long
using vec_int = vector<int>;
using vec_ii = vector<vec_int>;
using vec_iii = vector<vec_ii>;
using vec_iiii = vector<vec_iii>;
using P = pair<int,int>;
using T = tuple<int,int,int>;
using ll = long long;
using ld = long double;
#define rep(i, n) for(int i = 0; i < (int)(n); i++)

void cout_line(vector<int> &a){
    for(int i=0;i<a.size();i++){
        if(i<a.size()-1){
            cout<<a.at(i)<<" ";
        }else{
            cout<<a.at(i)<<endl;
        }
    }
}

int charToInt(char c){
    char zero_num = '0';
    return (int)c - (int)zero_num;
}

signed main(){
    int N; cin>>N;
    string S; cin>>S;
    int MOD = 998244353;

    if(N==1&&S=="?"){
        cout<<26<<endl;
        return 0;
    }else if(N==1){
        cout<<1<<endl;
        return 0;
    }

    if(N==2){
        if(S=="??"){
            cout<<26*25<<endl;
            return 0;
        }
        if(S.at(0)=='?'||S.at(1)=='?'){
            cout<<25<<endl;
            return 0;
        }else{
            if(S.at(0)!=S.at(1)){
                cout<<1<<endl;
                return 0;
            }else{
                cout<<0<<endl;
                return 0;
            }
        }
    }

    rep(i, S.size()-1){
        if(S.at(i)!='?'&&S.at(i+1)!='?'){
            if(S.at(i)==S.at(i+1)){
                cout<<0<<endl;
                return 0;
            }
        }
    }
    rep(i, S.size()-2){
        if(S.at(i)!='?'&&S.at(i+2)!='?'){
            if(S.at(i)==S.at(i+2)){
                cout<<0<<endl;
                return 0;
            }
        }
    }

    vec_iii DP(N-1, vec_ii(26, vec_int(26, 0)));
    rep(i, 26)rep(j, 26){
        // if(S.at(0)=='?')continue;
        // if(S.at(1)=='?')continue;
        if(S.at(0)!='?'&&(int)(S.at(0)-'a')!=i)continue;
        if(S.at(1)!='?'&&(int)(S.at(1)-'a')!=j)continue;
        if(i==j)continue;
        DP.at(0).at(i).at(j) = 1;
    }

    for(int i=1;i<N-1;i++){
        int num = (int)(S.at(i+1)-'a');
        int tot = 0;
        vec_int temp(26, 0);
        rep(j, 26)rep(k,26){
            temp.at(k) += DP.at(i-1).at(j).at(k);
        }

        rep(ii, 26){
            if(S.at(i+1)!='?'&&ii!=num)continue;
            rep(j, 26){
                if(ii==j)continue;
                DP.at(i).at(j).at(ii) += temp.at(j);
                DP.at(i).at(j).at(ii) -= DP.at(i-1).at(ii).at(j);
                DP.at(i).at(j).at(ii) += MOD;
                DP.at(i).at(j).at(ii) %= MOD;
            }
        }
        /*
        rep(ii, 26)rep(j, 26)rep(k,26){
            if(S.at(i+1)!='?'&&ii!=num)continue;
            if(ii==j||ii==k)continue;
            DP.at(i).at(j).at(ii) += DP.at(i-1).at(k).at(j);
            DP.at(i).at(j).at(ii) %= MOD;
        }
        */
    }
    int ans = 0;
    rep(i, 26) rep(j, 26){
        ans += DP.at(N-2).at(i).at(j);
        ans %= MOD;
    }
    cout<<ans<<endl;



    return 0;
}
0