結果

問題 No.1646 Avoid Palindrome
ユーザー HIcoderHIcoder
提出日時 2023-07-22 18:29:44
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 602 ms / 3,000 ms
コード長 2,151 bytes
コンパイル時間 1,284 ms
コンパイル使用メモリ 110,476 KB
実行使用メモリ 320,256 KB
最終ジャッジ日時 2024-09-22 18:01:34
合計ジャッジ時間 19,650 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 506 ms
301,824 KB
testcase_05 AC 501 ms
301,824 KB
testcase_06 AC 477 ms
291,712 KB
testcase_07 AC 494 ms
301,952 KB
testcase_08 AC 500 ms
304,640 KB
testcase_09 AC 476 ms
289,792 KB
testcase_10 AC 494 ms
296,832 KB
testcase_11 AC 474 ms
288,640 KB
testcase_12 AC 497 ms
301,312 KB
testcase_13 AC 502 ms
304,768 KB
testcase_14 AC 533 ms
296,064 KB
testcase_15 AC 541 ms
301,952 KB
testcase_16 AC 526 ms
294,144 KB
testcase_17 AC 550 ms
307,200 KB
testcase_18 AC 543 ms
299,008 KB
testcase_19 AC 533 ms
297,344 KB
testcase_20 AC 539 ms
299,776 KB
testcase_21 AC 552 ms
306,688 KB
testcase_22 AC 540 ms
296,448 KB
testcase_23 AC 571 ms
308,352 KB
testcase_24 AC 602 ms
320,256 KB
testcase_25 AC 594 ms
320,256 KB
testcase_26 AC 590 ms
320,128 KB
testcase_27 AC 592 ms
320,128 KB
testcase_28 AC 590 ms
320,128 KB
testcase_29 AC 514 ms
320,128 KB
testcase_30 AC 511 ms
320,256 KB
testcase_31 AC 512 ms
320,128 KB
testcase_32 AC 513 ms
320,256 KB
testcase_33 AC 513 ms
320,256 KB
testcase_34 AC 591 ms
320,256 KB
testcase_35 AC 2 ms
5,376 KB
testcase_36 AC 2 ms
5,376 KB
testcase_37 AC 2 ms
5,376 KB
testcase_38 AC 2 ms
5,376 KB
testcase_39 AC 2 ms
5,376 KB
testcase_40 AC 2 ms
5,376 KB
testcase_41 AC 1 ms
5,376 KB
testcase_42 AC 2 ms
5,376 KB
testcase_43 AC 508 ms
320,128 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

/*
O(N*σ^3)でTLE

*/
#include<iostream>
#include<set>
#include<algorithm>
#include<vector>
#include<string>
#include<set>
#include<map>
#include<stack>
#include<numeric>
#include<queue>
#include<cmath>
#include<deque>
using namespace std;
typedef long long ll;
const ll INF=1LL<<60;
typedef pair<int,int> P;
typedef pair<ll,P> PP;
const ll MOD=998244353;


int main(){
    int N;
    cin>>N;
    string S;
    cin>>S;

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

  

    vector dp(N+1,vector<vector<ll>>(26,vector<ll>(26,0)));

    vector<ll> sum(26,0);

    for(int j=0;j<26;j++){

        if( S[0]-'a' == j || S[0]=='?' ){

            for(int k=0;k<26;k++){
                if(j==k) continue;
                if(S[1]-'a'==k || S[1]=='?'){

                    dp[1][j][k]=1;
                    sum[k]+=1;
                    sum[k]%=MOD;

                }
            }
        }
    }

    for(int i=2;i<N;i++){

        for(int k=0;k<26;k++){

            /*
            dp[i][j][k]=i文字目がk, i-1文字目がj
            */

            if(S[i]!='?'){
                if(S[i]-'a'!=k){
                    //i=2のとき, i=0文字目がS[i]-'a'だった時を引けば良い
                    dp[i][k][S[i]-'a']+=(sum[k]-dp[i-1][S[i]-'a'][k]+MOD)%MOD;
                    dp[i][k][S[i]-'a']%=MOD;
                }
            }else{

                for(int l=0;l<26;l++){
                    //直前の文字列は順番に, j,k,l となっている. i文字目がl文字目
                    if(k!=l){
                        dp[i][k][l]+=(sum[k]-dp[i-1][l][k]+MOD)%MOD;
                        dp[i][k][l]%=MOD;
                    }                

                }


            }

           
        }

        for(int k=0;k<26;k++){
            sum[k]=0;
            for(int j=0;j<26;j++){
                sum[k]+=dp[i][j][k];
                sum[k]%=MOD;
            }
        }

    }

    ll ans=0;

    for(int j=0;j<26;j++){
        for(int k=0;k<26;k++){
            if(j==k) continue;
            ans+=dp[N-1][j][k];
            ans%=MOD;
        }
    }
    cout<<ans<<endl;


}
0