結果

問題 No.2708 Jewel holder
ユーザー Tatsu_mr
提出日時 2024-03-31 14:18:57
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 1,536 bytes
コンパイル時間 2,447 ms
コンパイル使用メモリ 211,924 KB
最終ジャッジ日時 2025-02-20 17:08:47
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

int main() {
    int h, w;
    cin >> h >> w;
    vector<vector<char>> a(h, vector<char>(w));
    for (int i = 0; i < h; i++) {
        for (int j = 0; j < w; j++) {
            cin >> a[i][j];
        }
    }
    //(i, j)でk個持ってる場合の数
    vector<vector<vector<long long>>> dp(h, vector<vector<long long>>(w, vector<long long>(110, 0LL)));
    vector<vector<vector<bool>>> dec(h, vector<vector<bool>>(w, vector<bool>(110, false)));
    dp[0][0][1] = 1LL;
    dec[0][0][1] = true;
    auto f = [&](auto f, int i, int j, int k) -> long long {
        if (!(0 <= i && i < h && 0 <= j && j < w && 0 <= k && k < 110)) {
            return 0LL;
        }
        if (dec[i][j][k]) {
            return dp[i][j][k];
        }
        if (a[i][j] == '#') {
            dec[i][j][k] = true;
            dp[i][j][k] = 0LL;
            return dp[i][j][k];
        } else if (a[i][j] == 'o') {
            if (k == 0) {
                dec[i][j][k] = true;
                dp[i][j][k] = 0LL;
                return dp[i][j][k];
            }
            dp[i][j][k] = f(f, i - 1, j, k - 1) + f(f, i, j - 1, k - 1);
            dec[i][j][k] = true;
            return dp[i][j][k];
        } else {
            dp[i][j][k] = f(f, i - 1, j, k + 1) + f(f, i, j - 1, k + 1);
            dec[i][j][k] = true;
            return dp[i][j][k];
        }
    };
    long long ans = 0LL;
    for (int k = 0; k < 110; k++) {
        ans += f(f, h - 1, w - 1, k);
    }
    cout << ans << endl;
}
0