結果

問題 No.2708 Jewel holder
コンテスト
ユーザー NP
提出日時 2024-03-31 15:02:23
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,353 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 341 ms
コンパイル使用メモリ 82,592 KB
実行使用メモリ 54,500 KB
最終ジャッジ日時 2024-09-30 20:11:54
合計ジャッジ時間 2,049 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 10 WA * 7
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

def count_valid_routes(H, W, grid):
    dp = [[0 for _ in range(W+1)] for _ in range(H+1)]
    dp[1][1] = 1
    gems = [[0 for _ in range(W+1)] for _ in range(H+1)]
    gems[1][1] = 1 if grid[0][0] == 'o' else 0
    if any('#' in row for row in grid):
        for i in range(1, H+1):
            for j in range(1, W+1):
                if i == 1 and j == 1:
                    continue
                if grid[i-1][j-1] == '#':
                    continue
                dp[i][j] = dp[i-1][j] + dp[i][j-1]
                gems[i][j] = max(gems[i-1][j], gems[i][j-1]) + (grid[i-1][j-1] == 'o') - (grid[i-1][j-1] == 'x')
                if gems[i][j] < 0:
                    dp[i][j] = 0
    else:
        for i in range(1, H+1):
            for j in range(1, W+1):
                if i == 1 and j == 1:
                    continue
                if grid[i-1][j-1] == '#':
                    continue
                if grid[i-1][j-1] == 'o':
                    gems[i][j] = max(gems[i-1][j], gems[i][j-1]) + 1
                else:
                    gems[i][j] = max(gems[i-1][j], gems[i][j-1]) - 1
                if gems[i][j] >= 0:
                    dp[i][j] = dp[i-1][j] + dp[i][j-1]
    return dp[-1][-1] if dp[-1][-1] >= 0 else 0
H, W = map(int, input().split())
grid = [input() for _ in range(H)]
print(count_valid_routes(H, W, grid))
0