몸과 마음이 건전한 SW 개발자

Softeer level3 우물 안 개구리 Python 본문

알고리즘

Softeer level3 우물 안 개구리 Python

스위태니 2023. 8. 28. 23:39

문제 링크 : https://softeer.ai/practice/info.do?idx=1&eid=394 

 

Softeer

연습문제를 담을 Set을 선택해주세요. 취소 확인

softeer.ai

정답 코드

import sys
input = sys.stdin.readline
N, M = map(int, input().split())
# N개의 역기
weightList = [0] + list(map(int, input().split()))
adjL = [[] for _ in range(N+1)]
for _ in range(M):
    start, to = map(int, input().split())
    adjL[start].append(to)
    adjL[to].append(start)

result = 0
for p in range(1, N+1):
    pWeight = weightList[p]
    imBest = True
    for r in adjL[p]:
        if weightList[r] >= pWeight:
             imBest = False
             break
    if imBest:
        result += 1

print(result)

느낀점

친분이 있는 회원이라길래 당연해 dfs인줄 알았는데 진짜 단순하게 서로 알고 있는 사이에서 자기가 제일 세다고 생각하는 회원의 수를 구하는 문제였다... 쉬운 문제를 빙빙 돌아갔다.