結果

問題 No.1602 With Animals into Institute 2
ユーザー shotoyoo
提出日時 2021-07-03 00:32:09
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
RE  
実行時間 -
コード長 2,857 bytes
コンパイル時間 128 ms
コンパイル使用メモリ 13,184 KB
実行使用メモリ 11,392 KB
最終ジャッジ日時 2024-07-02 02:08:19
合計ジャッジ時間 3,447 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample RE * 3
other RE * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda : sys.stdin.readline().rstrip()

sys.setrecursionlimit(2*10**5+10)
write = lambda x: sys.stdout.write(x+"\n")
debug = lambda x: sys.stderr.write(x+"\n")
writef = lambda x: print("{:.12f}".format(x))


class UF:
    # unionfind
    def __init__(self, n, depth):
        self.n = n
        self.depth = depth
        self.parent = list(range(n))
        self.size = [1] * n
    def check(self):
        return [self.root(i) for i in range(self.n)]
    def root(self, i):
        inter = set()
        while self.parent[i]!=i:
            inter.add(i)
            i = self.parent[i]
        r = i
        for i in inter:
            self.parent[i] = r
        return r
    def connect(self, i, j):
        # 繋いだかどうかを返す
        ri = self.root(i)
        rj = self.root(j)
        if ri==rj:
            return False
        if self.depth[rj]<self.depth[ri]:
            self.parent[ri] = rj
            self.size[rj] += self.size[ri]
        else:
            self.parent[rj] = ri
            self.size[ri] += self.size[rj]
        return True

def main3(ns,es,k):
    inf = 10**18
    n = len(ns)
    ans = [inf]*(n-1)
    import networkx as nx
    for s in range(n-1):
        gs = []
        for i in range(k):
            g = nx.Graph()
            for u in range(2*n):
                g.add_node(u)
            for u,v,c,x in es:
                vv = x>>i&1
                if not vv:
                    g.add_edge(u,n+v, weight=-c)
                    g.add_edge(v,n+u, weight=-c)
                else:
                    g.add_edge(u,v, weight=-c)
                    g.add_edge(n+v,n+u, weight=-c)
            for u in range(n-1):
                if u==s:
                    continue
                g.add_edge(u,n+u,weight=0)
            g.remove_node(n+s)
            g.remove_node(2*n-1)
            gs.append(g)
#             print(g.nodes)
#             print(g.edges)
        val = inf
        for g in gs:
            res = nx.max_weight_matching(g, maxcardinality=True)
            if len(res)<n-1:
                continue
            val = min(val, -sum((g[u][v]["weight"] for u,v in res)))
        if val==inf:
            val = -1
        ans[s] = val
    return ans
# グラフの読み込み・重みあり
inf = 10**16
n,m,k = map(int, input().split())
assert n<300
assert all((
    2<=n<=10**5,
    1<=m<=2*10**5,
    1<=k<=30,
))
ns = [[] for _ in range(n)]
es = []
for e in range(m):
    u,v,c,x = input().split()
    u = int(u)
    v = int(v)
    c = int(c)
    assert all((
        1<=u<=n,
        1<=v<=n,
        u!=v,
        1<=c<=10**9,
        len(x)==k,
        all((xx in("0","1") for xx in x))
    ))
    x = int(x, 2)
    u -= 1
    v -= 1
    ns[u].append((c,x,e,v))
    ns[v].append((c,x,e,u))
    es.append((u,v,c,x))
ans = main3(ns,es,k)
write("\n".join(map(str, ans)))
0