結果

問題 No.1646 Avoid Palindrome
ユーザー titia
提出日時 2021-08-14 00:46:39
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,101 bytes
コンパイル時間 188 ms
コンパイル使用メモリ 82,028 KB
実行使用メモリ 78,692 KB
最終ジャッジ日時 2024-10-04 02:10:52
合計ジャッジ時間 31,079 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 39 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

mod=998244353

N=int(input())
S=list(input().strip())

for i in range(N):
    S[i]=ord(S[i])-97

if N==1:
    print(1)
    exit()

DP=[0]*(26*26)

for i in range(26):
    for j in range(26):
        if i==j:
            continue

        if (i==S[0] or -34==S[0]) and (j==S[1] or -34==S[1]):
            DP[i*26+j]=1

#print(DP)

for stind in range(2,N):
    NDP1=[0]*(26*26)
    NDP2=[0]*(26*26)

    x=S[stind]

    for i in range(26):
        for j in range(26):
            kake=DP[i*26+j]

            if x==-34:
                NDP1[j*26]+=kake
                NDP1[j*26+i]-=kake
                if i!=25:
                    NDP1[j*26+i+1]+=kake
                NDP1[j*26+j]-=kake
                if j!=25:
                    NDP1[j*26+j+1]+=kake
                
            else:
                if i!=x and j!=x:
                    NDP2[j*26+x]+=kake

    for i in range(26):
        for j in range(1,26):
            NDP1[i*26+j]+=NDP1[i*26+j-1]

    DP=[0]*(26*26)

    for i in range(26*26):
        DP[i]=(NDP1[i]+NDP2[i])%mod

print(sum(DP)%mod)
0