結果

問題 No.3429 Palindromic Path (Hard)
コンテスト
ユーザー まみめ
提出日時 2025-12-29 23:43:27
言語 C++23
(gcc 15.2.0 + boost 1.89.0)
結果
AC  
実行時間 225 ms / 2,000 ms
コード長 2,388 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,821 ms
コンパイル使用メモリ 181,260 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2026-01-11 13:05:36
合計ジャッジ時間 2,630 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 7
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <vector>
#include <string>

using namespace std;

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    const long long MOD = 998244353;

    int N;
    cin >> N;

    vector<string> G(N);
    for (int i = 0; i < N; ++i)
    {
        cin >> G[i];
    }

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

    if (G[0][0] == G[N - 1][N - 1])
    {
        dp[1][N] = 1;
    }

    for (int k = 0; k < N - 1; ++k)
    {
        vector<vector<long long>> ndp(N + 1, vector<long long>(N + 1, 0));

        for (int r1 = 1; r1 <= N; ++r1)
        {
            int c1 = k + 2 - r1;
            if (!(1 <= c1 && c1 <= N))
            {
                continue;
            }

            for (int r2 = 1; r2 <= N; ++r2)
            {
                int c2 = 2 * N - k - r2;
                if (!(1 <= c2 && c2 <= N))
                {
                    continue;
                }

                long long cnt = dp[r1][r2];
                if (cnt == 0)
                {
                    continue;
                }

                vector<pair<int, int>> dir1 = {{1, 0}, {0, 1}};
                vector<pair<int, int>> dir2 = {{-1, 0}, {0, -1}};

                for (auto [dr1, dc1] : dir1)
                {
                    for (auto [dr2, dc2] : dir2)
                    {

                        int nr1 = r1 + dr1;
                        int nc1 = c1 + dc1;
                        int nr2 = r2 + dr2;
                        int nc2 = c2 + dc2;

                        if (!(1 <= nr1 && nr1 <= N && 1 <= nc1 && nc1 <= N))
                        {
                            continue;
                        }
                        if (!(1 <= nr2 && nr2 <= N && 1 <= nc2 && nc2 <= N))
                        {
                            continue;
                        }

                        if (G[nr1 - 1][nc1 - 1] == G[nr2 - 1][nc2 - 1])
                        {
                            ndp[nr1][nr2] = (ndp[nr1][nr2] + cnt) % MOD;
                        }
                    }
                }
            }
        }
        dp = ndp;
    }

    long long ans = 0;
    for (int r = 1; r <= N; ++r)
    {
        int c = N + 1 - r;
        if (1 <= c && c <= N)
        {
            ans = (ans + dp[r][r]) % MOD;
        }
    }

    cout << ans << "\n";

    return 0;
    }
0