結果

問題 No.2100 [Cherry Alpha C] Two-way Steps
ユーザー lloyz
提出日時 2022-10-15 14:00:18
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 356 ms / 2,000 ms
コード長 1,326 bytes
コンパイル時間 156 ms
コンパイル使用メモリ 82,064 KB
実行使用メモリ 129,640 KB
最終ジャッジ日時 2024-06-26 19:46:20
合計ジャッジ時間 13,261 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict
import sys
input = sys.stdin.readline

def main():
    n, m = map(int, input().split())
    H = list(map(int, input().split()))
    G = [defaultdict(list), defaultdict(list)]
    for _ in range(m):
        x, y = map(int, input().split())
        x -= 1
        y -= 1
        G[0][x].append((y, max(H[y] - H[x], 0)))
        G[1][y].append((x, max(H[x] - H[y], 0)))

    DP = [[[-1, -1] for _ in range(n)] for _ in range(2)]
    DP[0][0][0] = 0
    for i in range(n - 1):
        for j in range(2):
            if DP[0][i][j] != -1:
                for k, d in G[0][i]:
                    if d > 0:
                        if j == 0:
                            DP[0][k][1] = max(DP[0][k][1], DP[0][i][j] + d)
                    else:
                        DP[0][k][0] = max(DP[0][k][0], DP[0][i][j])
    print(max(DP[0][n - 1]))
    DP[1][n - 1][0] = 0
    for i in range(n - 1, 0, -1):
        for j in range(2):
            if DP[1][i][j] != -1:
                for k, d in G[1][i]:
                    if d > 0:
                        if j == 0:
                            DP[1][k][1] = max(DP[1][k][1], DP[1][i][j] + d)
                    else:
                        DP[1][k][0] = max(DP[1][k][0], DP[1][i][j])
    print(max(DP[1][0]))

if __name__ == '__main__':
    main()
0