[Softeer][level3] 우물 안 개구리 (python)

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

Softeer

 

문제

풀이

그냥 조건가지고 구현함.

db를 1로 초기화해서(0번째 index제외) 직접 index에 해당하는 무게를 비교하여 작은 경우 0으로 변경

db에 저장된 관계들 중 1인 경우의 갯수 출력

 

코드

import sys

N, M = map(int, input().split())
weights = [0] + list(map(int, sys.stdin.readline().split()))
relations = [list(map(int,input().split())) for _ in range(M)]

db = [0] + [1] * N

for i in range(M) :
    c1_weight = weights[relations[i][0]]
    c2_weight = weights[relations[i][1]]
    if c1_weight > c2_weight :
        db[relations[i][1]] = 0
    elif c1_weight < c2_weight :
        db[relations[i][0]] = 0
    else :
        db[relations[i][0]] = db[relations[i][1]] = 0

print(db.count(1))