Submission #6901873
Source Code Expand
# -*- coding: utf-8 -*- import sys from heapq import heappush, heappop def input(): return sys.stdin.readline().strip() def list2d(a, b, c): return [[c] * b for i in range(a)] def list3d(a, b, c, d): return [[[d] * c for j in range(b)] for i in range(a)] def ceil(x, y=1): return int(-(-x // y)) def INT(): return int(input()) def MAP(): return map(int, input().split()) def LIST(): return list(map(int, input().split())) def Yes(): print('Yes') def No(): print('No') def YES(): print('YES') def NO(): print('NO') sys.setrecursionlimit(10 ** 9) INF = float('inf') MOD = 10 ** 9 + 7 class MinCostFlow: INF = 10**18 def __init__(self, N): self.N = N self.G = [[] for i in range(N)] def add_edge(self, fr, to, cap, cost): G = self.G G[fr].append([to, cap, cost, len(G[to])]) G[to].append([fr, 0, -cost, len(G[fr])-1]) def flow(self, s, t, f): N = self.N; G = self.G INF = MinCostFlow.INF res = 0 H = [0]*N prv_v = [0]*N prv_e = [0]*N while f: dist = [INF]*N dist[s] = 0 que = [(0, s)] while que: c, v = heappop(que) if dist[v] < c: continue for i, (w, cap, cost, _) in enumerate(G[v]): if cap > 0 and dist[w] > dist[v] + cost + H[v] - H[w]: dist[w] = r = dist[v] + cost + H[v] - H[w] prv_v[w] = v; prv_e[w] = i heappush(que, (r, w)) if dist[t] == INF: return -1 for i in range(N): H[i] += dist[i] d = f; v = t while v != s: d = min(d, G[prv_v[v]][prv_e[v]][1]) v = prv_v[v] f -= d res += d * H[t] v = t while v != s: e = G[prv_v[v]][prv_e[v]] e[1] -= d G[v][e[3]][1] += d v = prv_v[v] return res R, G, B = MAP() N = R + G + B # 最小費用流 mcf = MinCostFlow(1006) # 始点からRGBの移動前 mcf.add_edge(1001, 1002, R, 0) mcf.add_edge(1001, 1003, G, 0) mcf.add_edge(1001, 1004, B, 0) # 0~1000の頂点番号は座標位置と一致させてある for v in range(1001): # R => 各座標 mcf.add_edge(1002, v, 1, abs(400-v)) # G => 各座標 mcf.add_edge(1003, v, 1, abs(500-v)) # B => 各座標 mcf.add_edge(1004, v, 1, abs(600-v)) # 各座標 => 終点 mcf.add_edge(v, 1005, 1, 0) print(mcf.flow(1001, 1005, N))
Submission Info
Submission Time | |
---|---|
Task | D - マーブル |
User | Coki628 |
Language | Python (3.4.3) |
Score | 40 |
Code Size | 2712 Byte |
Status | TLE |
Exec Time | 2104 ms |
Memory | 4596 KB |
Judge Result
Set Name | sub1 | sub2 | All | ||||||||
---|---|---|---|---|---|---|---|---|---|---|---|
Score / Max Score | 10 / 10 | 30 / 30 | 0 / 60 | ||||||||
Status |
|
|
|
Set Name | Test Cases |
---|---|
sub1 | sample_01_ABC.txt, test_ABC_01.txt, test_ABC_02.txt, test_ABC_03.txt, test_ABC_04.txt, test_ABC_05.txt, test_ABC_06.txt, test_ABC_07.txt, test_ABC_08.txt, test_ABC_09.txt, test_ABC_10.txt, test_ABC_11.txt, test_ABC_12.txt, test_ABC_13.txt, test_ABC_14.txt, test_ABC_15.txt, test_ABC_16.txt, test_ABC_17.txt, test_ABC_18.txt, test_ABC_19.txt, test_ABC_20.txt, test_ABC_21.txt, test_ABC_22.txt, test_ABC_23.txt, test_ABC_24.txt, test_ABC_25.txt, test_ABC_26.txt, test_ABC_27.txt, test_ABC_28.txt |
sub2 | sample_01_ABC.txt, sample_02_BC.txt, test_ABC_01.txt, test_ABC_02.txt, test_ABC_03.txt, test_ABC_04.txt, test_ABC_05.txt, test_ABC_06.txt, test_ABC_07.txt, test_ABC_08.txt, test_ABC_09.txt, test_ABC_10.txt, test_ABC_11.txt, test_ABC_12.txt, test_ABC_13.txt, test_ABC_14.txt, test_ABC_15.txt, test_ABC_16.txt, test_ABC_17.txt, test_ABC_18.txt, test_ABC_19.txt, test_ABC_20.txt, test_ABC_21.txt, test_ABC_22.txt, test_ABC_23.txt, test_ABC_24.txt, test_ABC_25.txt, test_ABC_26.txt, test_ABC_27.txt, test_ABC_28.txt, test_BC_29.txt, test_BC_30.txt, test_BC_31.txt, test_BC_32.txt, test_BC_33.txt, test_BC_34.txt, test_BC_35.txt, test_BC_36.txt, test_BC_37.txt, test_BC_38.txt, test_BC_39.txt, test_BC_40.txt, test_BC_41.txt, test_BC_42.txt, test_BC_43.txt, test_BC_44.txt, test_BC_45.txt, test_BC_46.txt, test_BC_47.txt, test_BC_48.txt, test_BC_49.txt, test_BC_50.txt, test_BC_51.txt, test_BC_52.txt, test_BC_53.txt, test_BC_54.txt, test_BC_55.txt |
All | sample_01_ABC.txt, sample_02_BC.txt, sample_03_C.txt, test_ABC_01.txt, test_ABC_02.txt, test_ABC_03.txt, test_ABC_04.txt, test_ABC_05.txt, test_ABC_06.txt, test_ABC_07.txt, test_ABC_08.txt, test_ABC_09.txt, test_ABC_10.txt, test_ABC_11.txt, test_ABC_12.txt, test_ABC_13.txt, test_ABC_14.txt, test_ABC_15.txt, test_ABC_16.txt, test_ABC_17.txt, test_ABC_18.txt, test_ABC_19.txt, test_ABC_20.txt, test_ABC_21.txt, test_ABC_22.txt, test_ABC_23.txt, test_ABC_24.txt, test_ABC_25.txt, test_ABC_26.txt, test_ABC_27.txt, test_ABC_28.txt, test_BC_29.txt, test_BC_30.txt, test_BC_31.txt, test_BC_32.txt, test_BC_33.txt, test_BC_34.txt, test_BC_35.txt, test_BC_36.txt, test_BC_37.txt, test_BC_38.txt, test_BC_39.txt, test_BC_40.txt, test_BC_41.txt, test_BC_42.txt, test_BC_43.txt, test_BC_44.txt, test_BC_45.txt, test_BC_46.txt, test_BC_47.txt, test_BC_48.txt, test_BC_49.txt, test_BC_50.txt, test_BC_51.txt, test_BC_52.txt, test_BC_53.txt, test_BC_54.txt, test_BC_55.txt, test_C_56.txt, test_C_57.txt, test_C_58.txt, test_C_59.txt, test_C_60.txt, test_C_61.txt, test_C_62.txt, test_C_63.txt, test_C_64.txt, test_C_65.txt, test_C_66.txt, test_C_67.txt, test_C_68.txt, test_C_69.txt, test_C_70.txt, test_C_71.txt, test_C_72.txt, test_C_73.txt, test_C_74.txt, test_C_75.txt, test_C_76.txt, test_C_77.txt, test_C_78.txt, test_C_79.txt, test_C_80.txt, test_C_81.txt, test_C_82.txt |
Case Name | Status | Exec Time | Memory |
---|---|---|---|
sample_01_ABC.txt | AC | 59 ms | 4596 KB |
sample_02_BC.txt | AC | 213 ms | 4596 KB |
sample_03_C.txt | TLE | 2104 ms | 4596 KB |
test_ABC_01.txt | AC | 58 ms | 4596 KB |
test_ABC_02.txt | AC | 61 ms | 4596 KB |
test_ABC_03.txt | AC | 61 ms | 4596 KB |
test_ABC_04.txt | AC | 69 ms | 4596 KB |
test_ABC_05.txt | AC | 50 ms | 4596 KB |
test_ABC_06.txt | AC | 59 ms | 4596 KB |
test_ABC_07.txt | AC | 62 ms | 4596 KB |
test_ABC_08.txt | AC | 66 ms | 4596 KB |
test_ABC_09.txt | AC | 55 ms | 4596 KB |
test_ABC_10.txt | AC | 57 ms | 4596 KB |
test_ABC_11.txt | AC | 76 ms | 4596 KB |
test_ABC_12.txt | AC | 54 ms | 4596 KB |
test_ABC_13.txt | AC | 54 ms | 4596 KB |
test_ABC_14.txt | AC | 60 ms | 4596 KB |
test_ABC_15.txt | AC | 56 ms | 4596 KB |
test_ABC_16.txt | AC | 60 ms | 4596 KB |
test_ABC_17.txt | AC | 70 ms | 4596 KB |
test_ABC_18.txt | AC | 69 ms | 4596 KB |
test_ABC_19.txt | AC | 68 ms | 4596 KB |
test_ABC_20.txt | AC | 59 ms | 4596 KB |
test_ABC_21.txt | AC | 37 ms | 4596 KB |
test_ABC_22.txt | AC | 54 ms | 4596 KB |
test_ABC_23.txt | AC | 52 ms | 4596 KB |
test_ABC_24.txt | AC | 50 ms | 4596 KB |
test_ABC_25.txt | AC | 66 ms | 4596 KB |
test_ABC_26.txt | AC | 68 ms | 4596 KB |
test_ABC_27.txt | AC | 70 ms | 4596 KB |
test_ABC_28.txt | AC | 84 ms | 4596 KB |
test_BC_29.txt | AC | 155 ms | 4596 KB |
test_BC_30.txt | AC | 156 ms | 4596 KB |
test_BC_31.txt | AC | 242 ms | 4596 KB |
test_BC_32.txt | AC | 255 ms | 4596 KB |
test_BC_33.txt | AC | 327 ms | 4596 KB |
test_BC_34.txt | AC | 327 ms | 4596 KB |
test_BC_35.txt | AC | 200 ms | 4596 KB |
test_BC_36.txt | AC | 273 ms | 4596 KB |
test_BC_37.txt | AC | 267 ms | 4596 KB |
test_BC_38.txt | AC | 266 ms | 4596 KB |
test_BC_39.txt | AC | 196 ms | 4596 KB |
test_BC_40.txt | AC | 346 ms | 4596 KB |
test_BC_41.txt | AC | 318 ms | 4596 KB |
test_BC_42.txt | AC | 405 ms | 4596 KB |
test_BC_43.txt | AC | 239 ms | 4596 KB |
test_BC_44.txt | AC | 192 ms | 4596 KB |
test_BC_45.txt | AC | 192 ms | 4596 KB |
test_BC_46.txt | AC | 322 ms | 4596 KB |
test_BC_47.txt | AC | 250 ms | 4596 KB |
test_BC_48.txt | AC | 385 ms | 4596 KB |
test_BC_49.txt | AC | 326 ms | 4596 KB |
test_BC_50.txt | AC | 317 ms | 4596 KB |
test_BC_51.txt | AC | 351 ms | 4596 KB |
test_BC_52.txt | AC | 163 ms | 4596 KB |
test_BC_53.txt | AC | 174 ms | 4596 KB |
test_BC_54.txt | AC | 193 ms | 4596 KB |
test_BC_55.txt | AC | 517 ms | 4596 KB |
test_C_56.txt | AC | 1116 ms | 4596 KB |
test_C_57.txt | AC | 1953 ms | 4596 KB |
test_C_58.txt | TLE | 2104 ms | 4596 KB |
test_C_59.txt | AC | 1778 ms | 4596 KB |
test_C_60.txt | AC | 900 ms | 4596 KB |
test_C_61.txt | AC | 1539 ms | 4596 KB |
test_C_62.txt | AC | 1442 ms | 4596 KB |
test_C_63.txt | AC | 1542 ms | 4596 KB |
test_C_64.txt | AC | 1254 ms | 4596 KB |
test_C_65.txt | AC | 1708 ms | 4596 KB |
test_C_66.txt | AC | 1647 ms | 4596 KB |
test_C_67.txt | AC | 1844 ms | 4596 KB |
test_C_68.txt | AC | 1015 ms | 4596 KB |
test_C_69.txt | AC | 1219 ms | 4596 KB |
test_C_70.txt | AC | 1465 ms | 4596 KB |
test_C_71.txt | AC | 1948 ms | 4596 KB |
test_C_72.txt | TLE | 2104 ms | 4596 KB |
test_C_73.txt | TLE | 2031 ms | 4596 KB |
test_C_74.txt | AC | 1946 ms | 4596 KB |
test_C_75.txt | AC | 1844 ms | 4596 KB |
test_C_76.txt | TLE | 2103 ms | 4596 KB |
test_C_77.txt | TLE | 2104 ms | 4596 KB |
test_C_78.txt | TLE | 2104 ms | 4596 KB |
test_C_79.txt | AC | 1004 ms | 4596 KB |
test_C_80.txt | AC | 1066 ms | 4596 KB |
test_C_81.txt | AC | 1177 ms | 4596 KB |
test_C_82.txt | TLE | 2104 ms | 4596 KB |