結果

問題 No.654 Air E869120
ユーザー りんりん
提出日時 2023-07-09 16:58:33
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 2,913 bytes
コンパイル時間 744 ms
コンパイル使用メモリ 82,396 KB
実行使用メモリ 191,488 KB
最終ジャッジ日時 2024-07-23 14:12:48
合計ジャッジ時間 8,697 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 41 ms
58,368 KB
testcase_01 AC 38 ms
52,608 KB
testcase_02 AC 39 ms
53,120 KB
testcase_03 AC 37 ms
53,120 KB
testcase_04 AC 41 ms
52,864 KB
testcase_05 AC 40 ms
53,120 KB
testcase_06 AC 39 ms
52,864 KB
testcase_07 AC 38 ms
53,212 KB
testcase_08 AC 38 ms
52,608 KB
testcase_09 AC 38 ms
53,248 KB
testcase_10 AC 998 ms
83,472 KB
testcase_11 AC 595 ms
80,960 KB
testcase_12 AC 627 ms
80,800 KB
testcase_13 AC 896 ms
82,896 KB
testcase_14 AC 534 ms
80,220 KB
testcase_15 AC 595 ms
81,092 KB
testcase_16 TLE -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
testcase_39 -- -
権限があれば一括ダウンロードができます

ソースコード

diff #

#https://yukicoder.me/problems/no/654
#TLEしてるけどアルゴリズムは合ってるはず
import sys

sys.setrecursionlimit(10**7)
class FordFulkerson:
    def __init__(self,V):
        self.v = V
        self.edges = [[] for _ in range(V)] #隣接リスト
    
    def addEdge(self,fr,to,cap):
        forward = [to,cap,None] #行き先、capasity,逆向きの辺
        forward[2] = backward = [fr,0,forward]
        self.edges[fr].append(forward)
        self.edges[to].append(backward)
    
    def dfs(self,v,t,f): #tはdestination,fは追加できる最大のフロー量
        if v == t:
            return f
        used = self.used
        used[v] = 1
        for e in self.edges[v]:
            w,cap,rev = e
            if cap and not used[w]: #辺がありかつまだwを見ていない時
                d = self.dfs(w,t,min(f,cap))
                if d: #結局、図26.6を見ればわかるけど、道pとして選んだところのcapを減らし、その逆の道のcapを増やしている。
                    e[1] -= d
                    rev[1] += d
                    return d
        return 0 #辿れる道がなかったら0を返さなきゃいけないからこれは必要。
    
    def flow(self,s,t):
        flow = 0
        f = INF = float('inf')
        v = self.v
        while f: #ここは最大O(Fmax)
            self.used = [0]*v #used初期化
            f = self.dfs(s,t,INF) #これはO(E'+V) = O(E) (まず、グラフは連結(p297)なのでE>V、そしてE'<=2*Eよりこれが成り立つ)
            flow += f
        return flow
            
N,M,d = map(int,input().split())

#(u,pi)→(v,qi)にcapacityがwiの辺を設定
#(u,pi)→(u,pi+1)には容量無限大の辺を設定(空港で待つことを意味する)

#上の方法だと間に合わないから
#辺を貼るのは街iに飛行機が到着する時刻だけでいい。
def convert(u,time):
    return u*maxtime + time_dict[time]
    
INF = float('inf')

#dを考慮する。Aに出発しBに到着する飛行機はAに出発し、B+dに到着するとみなすことができる。
events = []
time = set()
for _ in range(M):
    u,v,p,q,w = map(int,input().split()) #このqは1日の終わりを超えることはない。
    u -= 1
    v -= 1
    q += d
    #(u,p)→(v,q+d)へcapacityがwの辺
    events.append((u,v,p,q,w))
    time.add(p)
    time.add(q)

time = sorted(list(time))
maxtime = len(time)
time_dict = {t:i for i,t in enumerate(time)}
ff = FordFulkerson(maxtime*N)
wait = [[] for _ in range(N)]
for u,v,p,q,w in events:
    ff.addEdge(convert(u,p),convert(v,q),w)
    wait[u].append(p)
    wait[v].append(q)
    
for i in range(N):
    wait[i].sort()
    for j in range(len(wait[i])-1):
        t1 = wait[i][j]
        t2 = wait[i][j+1]
        ff.addEdge(convert(i,t1),convert(i,t2),INF)

print(ff.flow(convert(0,wait[0][0]),convert(N-1,wait[N-1][-1])))
0