結果
問題 | No.134 走れ!サブロー君 |
ユーザー |
|
提出日時 | 2024-03-15 15:52:22 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 167 ms / 5,000 ms |
コード長 | 1,725 bytes |
コンパイル時間 | 193 ms |
コンパイル使用メモリ | 82,428 KB |
実行使用メモリ | 80,000 KB |
最終ジャッジ日時 | 2024-09-30 00:02:02 |
合計ジャッジ時間 | 2,175 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 15 |
ソースコード
import sysimport mathimport bisectfrom heapq import heapify, heappop, heappushfrom collections import deque, defaultdict, Counterfrom functools import lru_cachefrom itertools import accumulate, combinations, permutations, productsys.setrecursionlimit(1000000)MOD = 10 ** 9 + 7MOD99 = 998244353input = lambda: sys.stdin.readline().strip()NI = lambda: int(input())NMI = lambda: map(float, input().split())NLI = lambda: list(NMI())SI = lambda: input()SMI = lambda: input().split()SLI = lambda: list(SMI())EI = lambda m: [NLI() for _ in range(m)]def TSP(N, D, start, W):"""巡回セールスマン問題 O(N^2*2^N):param N: 頂点数:param D: NxNの距離行列:param start: 始点のindex:return: 最短距離"""INF = 10**15dp = [[INF] * (1<<N) for _ in range(N)]dp[start][0] = 0for case in range(1<<N):w = 0for i in range(N):if (case >> i) & 1 == 0:w += W[i]for u in range(N):if dp[u][case] >= INF/2:continuefor v in range(N):if (case >> v) & 1:continuedp[v][case|(1<<v)] = min(dp[v][case|(1<<v)], dp[u][case] + D[u][v] * (w+100) / 120 + W[u])return dp[start][-1]def main():XY = [NLI()]N = NI()W = [0]for _ in range(N):x, y, w = NMI()XY.append([x, y])W.append(w)D = [[0]*(N+1) for _ in range(N+1)]for i in range(N+1):xi, yi = XY[i]for j in range(N+1):xj, yj = XY[j]D[i][j] = abs(xi-xj) + abs(yi-yj)ans = TSP(N+1, D, 0, W)print(ans)if __name__ == "__main__":main()