結果

問題 No.904 サメトロ
ユーザー NoneNone
提出日時 2021-04-02 23:23:59
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 5,768 bytes
コンパイル時間 433 ms
コンパイル使用メモリ 86,744 KB
実行使用メモリ 112,776 KB
最終ジャッジ日時 2023-08-25 13:53:37
合計ジャッジ時間 10,323 ms
ジャッジサーバーID
(参考情報)
judge12 / judge14
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 241 ms
82,340 KB
testcase_01 WA -
testcase_02 AC 271 ms
83,768 KB
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 AC 88 ms
71,732 KB
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 AC 79 ms
71,548 KB
testcase_19 WA -
testcase_20 AC 182 ms
81,040 KB
testcase_21 WA -
testcase_22 AC 321 ms
90,896 KB
testcase_23 WA -
testcase_24 AC 194 ms
80,316 KB
testcase_25 WA -
testcase_26 WA -
testcase_27 AC 421 ms
112,776 KB
testcase_28 AC 80 ms
71,392 KB
testcase_29 AC 86 ms
76,460 KB
testcase_30 WA -
testcase_31 WA -
testcase_32 AC 80 ms
71,304 KB
testcase_33 AC 81 ms
71,392 KB
testcase_34 AC 80 ms
71,600 KB
testcase_35 AC 105 ms
77,640 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

class MaxFlow:
    def __init__(self, n=0):
        self._n = n
        self.g = [[] for _ in range(n)]
        self.pos = []

    def add_edge(self, frm, to, cap):
        m = len(self.pos)
        e1 = MaxFlow._edge(to, cap)
        e2 = MaxFlow._edge(frm, 0)
        e1.rev = e2
        e2.rev = e1
        self.pos.append(e1)
        self.g[frm].append(e1)
        self.g[to].append(e2)
        return m

    class edge:
        def __init__(self, frm, to, cap, flow):
            self.frm = frm
            self.to = to
            self.cap = cap
            self.flow = flow

        def __iter__(self):
            yield self.frm
            yield self.to
            yield self.cap
            yield self.flow

    def get_edge(self, i):
        """ i 番目に追加された辺について (from, to, 初期の cap, 現在の流量(始めは 0)) """
        e1 = self.pos[i]
        e2 = e1.rev
        return MaxFlow.edge(e2.to, e1.to, e1.cap + e2.cap, e2.cap)

    def edges(self):
        """ get_edge の返り値をリストで並べたもの """
        return [self.get_edge(i) for i in range(len(self.pos))]

    def change_edge(self, i, new_cap, new_flow):
        e = self.pos[i]
        e.cap = new_cap - new_flow
        e.rev.cap = new_flow

    def flow(self, s, t, flow_limit=0XFFFFFFFFFFFFFFF):
        g = self.g
        flow = 0
        while flow < flow_limit:
            level = [-1] * self._n
            level[s] = 0
            que = [None] * self._n
            ql = 0
            qr = 1
            que[0] = s
            unreached = True
            while unreached and ql < qr:
                v = que[ql]
                ql += 1
                for e in g[v]:
                    to = e.to
                    if e.cap and level[to] < 0:
                        level[to] = level[v] + 1
                        if to == t:
                            unreached = False
                            break
                        que[qr] = to
                        qr += 1
            if unreached:
                return flow
            ptr = [len(es) for es in g]
            stack = []
            v = t
            up = flow_limit - flow
            res = 0
            while True:
                if v == s or not ptr[v]:
                    if v == s:
                        res = up
                    while stack:
                        tmp = res
                        e, up, res = stack.pop()
                        e.cap -= tmp
                        e.rev.cap += tmp
                        res += tmp
                        if res < up:
                            v = e.to
                            break
                    else:
                        flow += res
                        break
                i = ptr[v]
                while i:
                    i -= 1
                    e = g[v][i]
                    if level[e.to] == level[v] - 1 and e.rev.cap:
                        ptr[v] = i
                        stack.append((e.rev, up, res))
                        v = e.to
                        up = min(up, e.rev.cap)
                        res = 0
                        break
                else:
                    ptr[v] = i
        return flow

    def min_cut(self, s):
        """
        残余グラフから到達可能な頂点集合を返す。
        流量の最大値を十分大きくとった場合、この頂点集合はmin_cutとなる。
        """
        visited = [False] * self._n
        que = [None] * self._n
        ql = 0
        qr = 1
        que[0] = s
        visited[s] = True
        while ql < qr:
            p = que[ql]
            ql += 1
            for e in self.g[p]:
                if e.cap and not visited[e.to]:
                    visited[e.to] = True
                    que[qr] = e.to
                    qr += 1
        return visited

    def draw(self):
        """
        :return: グラフを可視化
        """
        import matplotlib.pyplot as plt
        import networkx as nx

        G = nx.DiGraph()

        for frm, to, cap, cap_now in self.edges():
            G.add_edge(frm, to, label="{}/{}".format(cap_now,cap))


        edge_labels = {(i, j): w['label'] for i, j, w in G.edges(data=True)}
        pos = nx.spring_layout(G)
        nx.draw_networkx(G, pos, with_labels=True, connectionstyle='arc3, rad = 0.1')
        nx.draw_networkx_edge_labels(G, pos, edge_labels=edge_labels)
        plt.axis("off")
        plt.show()

    class _edge:
        def __init__(self, to, cap):
            self.to = to
            self.cap = cap

        def __iter__(self):
            yield self.to
            yield self.cap


def binary_search_int(ok, ng, test):
    """
    :param ok: solve(x) = True を必ず満たす点
    :param ng: solve(x) = False を必ず満たす点
    """
    while abs(ok - ng) > 1:
        mid = (ok + ng) // 2
        if test(mid):
            ok = mid
        else:
            ng = mid
    return ok

##############################################################

import sys
input = sys.stdin.readline




# example()

INF=10**2

N=int(input())

mf = MaxFlow(2*N+2)

SA,SB=0,0
for i in range(1,N):
    a,b=map(int, input().split())
    for j in range(N):
        if i==j:continue
        mf.add_edge(i,N+j,a)
    mf.add_edge(2*N,i,a)
    mf.add_edge(N+i,2*N+1,b)
    SA+=a
    SB+=b

for i in range(N):
    mf.add_edge(0,N+i,INF)

mf.add_edge(N,2*N+1,INF)

from copy import deepcopy

def test(x):
    mf2=deepcopy(mf)
    mf2.add_edge(2*N,0,x)
    mf2.flow(2*N,2*N+1)
    flg=1
    for i in range(2*N):
        if i==0 or i==N: continue
        frm,to,cap,now=mf.get_edge(i)
        if now!=0:
            flg=0
    return flg

a=binary_search_int(SB,-1,test)

print(SB-a+1)
0