結果

問題 No.1 道のショートカット
ユーザー warashi
提出日時 2015-02-08 17:45:31
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,274 bytes
コンパイル時間 570 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 553,592 KB
最終ジャッジ日時 2024-07-08 03:45:07
合計ジャッジ時間 7,304 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 WA * 1
other AC * 6 WA * 1 MLE * 1 -- * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3
import functools
import operator


def memoize(fn):
    table = {}

    def func(*args):
        if args not in table:
            table[args] = fn(*args)
        return table[args]
    return func

N = int(input())
C = int(input())
V = int(input())
S = [int(s) for s in input().split()]
T = [int(t) for t in input().split()]
Y = [int(y) for y in input().split()]
M = [int(m) for m in input().split()]

PATH = list(zip(S, T))
COST = dict(zip(PATH, Y))
METER = dict(zip(PATH, M))


@memoize
def path(town, yen):
    if town == 1:
        return [[1]]
    road = [path(s, yen - COST[s, t])
            for s, t in PATH if t == town and yen >= COST[s, t]]
    if len(road) < 1:
        return [[-1]]
    return [p + [town] for p in functools.reduce(operator.add, road)]


def cost(path):
    cost = 0
    for i in range(1, len(path)):
        if path[i - 1] < 0:
            continue
        cost += COST[path[i - 1], path[i]]
    return cost


def meter(path):
    meter = 0
    for i in range(1, len(path)):
        if path[i - 1] < 0:
            continue
        meter += METER[path[i - 1], path[i]]
    return meter


route = [meter(p) for p in path(N, C) if p[0] == 1 and cost(p) <= C]
if len(route) == 0:
    print(-1)
else:
    print(min(route))
0