백준/ Gold 4 문제 , 백준 파이썬 14621 , 나만 안되는 연애 [최소 스패닝 트리]

백준/ Gold 4 문제 , 백준 파이썬 14621 , 나만 안되는 연애 [최소 스패닝 트리]

 

 

풀이 시간

 

 

Check Point !   ( 해당사항 ✓체크 )

1. 막힘 없이 수월하게 풀린 문제인가?

2. 1시간이내로 풀렸던 문제인가?

3. 1시간 이상 or 며칠을 두고 풀어봤더니 풀린 문제인가?

4. 시간을 써도 도무지 풀 수 없는 문제인가?

5. 솔루션을 찾아봤는가?

-------------------------------------------------------------------------------------------
난이도 체감
1. 최상
2. 상
3. 중
4. 하


<이해도>
1. 완벽히 이해
2. 다소 헷갈리는 부분들이 있음
3. 이해 못함


<덧붙일 말>
남자학교에서는 여자학교만 갈수 있고 여자학교에서는 남자학교만 갈 수 있다는 점을 이용하여 입력 받은 걸 모두 넣지 않고 간선이 서로 다른 성별의 학교로 이어져있을때만 집어 넣는다.

 

 

문제 출처

https://www.acmicpc.net/problem/14621

14621번: 나만 안되는 연애

 

 

풀이

 

 

 

입력된 간선이 성별이 다른 학교인지 확인해주고 넣어준다.

 

 

 

정답

 

 

 

import heapq
import sys
input=sys.stdin.readline

N,M=map(int,input().split())
school=['X']+list(input().split())
graph=[[] for _ in range(N+1)]

for i in range(M):
    u,v,d=map(int,input().split())
    if school[u]!=school[v]:
        graph[u].append((d,v))
        graph[v].append((d,u))

q=graph[1]
visited=[True]+[False]*N
visited[1]=True
cnt=0
heapq.heapify(q)
result=0
while q:
    Z,X=heapq.heappop(q)
    if not visited[X]:
        visited[X]=True
        cnt+=1
        result+=Z
        for i in graph[X]:
            if not visited[i[1]]:
                heapq.heappush(q,i)
if cnt==N-1:
    print(result)
else:
    print(-1)