結果

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

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/modint>

using namespace std;
using namespace atcoder;

using mint = modint998244353;

int main()
{
    int n;
    string s;
    cin >> n >> s;

    vector dp(n + 1, vector(27, vector<mint>(27)));
    dp.at(0).at(26).at(26) = 1;
    for (int i = 0; i < n; i++)
    {
        if (s.at(i) == '?')
        {
            for (int k = 0; k < 27; k++)
            {
                mint s = 0;
                for (int j = 0; j < 27; j++)
                {
                    s += dp.at(i).at(j).at(k);
                }

                for (int l = 0; l < 26; l++)
                {
                    if (l == k)
                    {
                        continue;
                    }
                    dp.at(i + 1).at(k).at(l) += s - dp.at(i).at(l).at(k);
                }
            }
            continue;
        }

        for (int j = 0; j < 27; j++)
        {
            for (int k = 0; k < 27; k++)
            {
                int l = s.at(i) - 'a';
                if (l == j or l == k)
                {
                    continue;
                }
                dp.at(i + 1).at(k).at(l) += dp.at(i).at(j).at(k);
            }
        }
    }

    mint ans = 0;
    for (int i = 0; i < 27; i++)
    {
        for (int j = 0; j < 27; j++)
        {
            ans += dp.at(n).at(i).at(j);
        }
    }
    cout << ans.val() << endl;
    return 0;
}
0