[softeer] 소프티어 우물 안 개구리 파이썬 python

hyewon9913·2024년 6월 27일
0

코딩테스트(python)

목록 보기
34/46
post-thumbnail

문제

처음 이 문제를 보고 2차원 리스트를 활용하여 i번째 리스트에는 i번 회원의 친구들의 무게를 담아줘서 그 리스트의 max 값이 i번 회원이 들 수 있는 무게보다 작다면 answer에 +1 해줘야겠다는 생각을 했다.

i번 회원이 들 수 있는 무게 > max( i번 회원의 친구들이 들 수 있는 무게가 담긴 리스트 )

정리하자면 이때 answer값이 +1 되는 것이다.

코드

import sys


n, m = map(int,input().split())
w = list(map(int,input().split()))

#각 회원의 친구들의 무게리스트를 저장할 리스트
ans = [[0] for _ in range(n+1)]


for i in range(m):
    a,b = map(int,input().split())
    #a의 친구 b의 무게
    ans[a].append(w[b-1])
    #b의 친구 a의 무게
    ans[b].append(w[a-1])

answer = 0
for i in range(1,n+1):
    if w[i-1] > max(ans[i]):
        answer+=1

print(answer)

이때 인덱스 값과 회원번호는 1이 차이나므로 이 부분에 대한 처리를 유의해서 해줘야한다.

profile
차근차근 굴러가는 코딩일지

0개의 댓글