1948G - MST with Matching - CodeForces Solution


bitmasks brute force dsu trees

Please click on ads to support us..

Python Code:

def get_root(v, fa):
    if fa[v] == v: return v
    fa[v] = get_root(fa[v], fa)
    return fa[v]
n, c = map(int, input().split())
e = [list(map(int, input().split())) for _ in range(n)]
edge = [(e[i][j], i, j) for i in range(n) for j in range(i + 1, n) if e[i][j] > 0]
edge.sort()
inf = float('inf'); ans = inf
for i in range(1, 1 << n):
    now = bin(i).count('1') * c; fa = list(range(n)); tot = n
    for w, u, v in edge:
        if now > ans: break
        if ((i >> u) & 1) or ((i >> v) & 1):
            u = get_root(u, fa); v = get_root(v, fa)
            if u == v: pass
            else: fa[u] = v; now += w; tot -= 1
    if tot == 1: ans = min(ans, now)
print(ans)


Comments

Submit
0 Comments
More Questions

1515E - Phoenix and Computers
1552B - Running for Gold
994A - Fingerprints
1221C - Perfect Team
1709C - Recover an RBS
378A - Playing with Dice
248B - Chilly Willy
1709B - Also Try Minecraft
1418A - Buying Torches
131C - The World is a Theatre
1696A - NIT orz
1178D - Prime Graph
1711D - Rain
534A - Exam
1472A - Cards for Friends
315A - Sereja and Bottles
1697C - awoo's Favorite Problem
165A - Supercentral Point
1493A - Anti-knapsack
1493B - Planet Lapituletti
747B - Mammoth's Genome Decoding
1591C - Minimize Distance
1182B - Plus from Picture
1674B - Dictionary
1426C - Increase and Copy
520C - DNA Alignment
767A - Snacktower
1365A - Matrix Game
714B - Filya and Homework
31A - Worms Evolution