소프티어 | 우물 안 개구리 ⭐⭐⭐ - PYTHON

소프티어 | 우물 안 개구리 ⭐⭐⭐ - PYTHON

 

문제

https://softeer.ai/practice/6289

Softeer - 현대자동차그룹 SW인재확보플랫폼

 

 

🤖 알고리즘

가장 먼저 첫 번째 주어진 값(f)와 두 번째 주어진 값(s) 을 인덱스-1 값으로 생각하여

f가 클 때, s가 클 때, f와  s가 같을 때 세 경우로 나누어 고민하였다.

 

이 후, 1로 채운 리스트를 생성하고 위 세 경우일 때 0으로 변환시켰고, 1인 경우의 수를 출력하였다.

 

✏️ 내 풀이

 

첫 풀이

import sys
input = sys.stdin.readline

N, M = map(int, input().split())
members = list(map(int, input().split()))
data = [list(map(int, input().split())) for _ in range(M)]

cnt_list = [1]*N
result = 0

for i in range(M):
    if members[data[i][0] -1] < members[data[i][1] -1]:
        cnt_list[data[i][0] -1] = 0
    elif members[data[i][0] -1] == members[data[i][1] -1]:
        cnt_list[data[i][0]-1] = 0
        cnt_list[data[i][1]-1] = 0
    elif members[data[i][0] -1] > members[data[i][1] -1]:       
        cnt_list[data[i][1]-1] = 0

for i in range(N):
    if cnt_list[i] > 0:
        result += 1

print(result)

 

제출을 하고 나니 코드가 마음에 들지 않았다

 

⭐ 보완한 점

✅ 지저분 한 코드 정리

✅ 굳이 result 변수를 이용하지 않고, 리스트에서 count 하기

✅ data 값을 미리 받지 않고 반복문 안에서 각각 input 하기

 

 

최종 풀이

import sys
input = sys.stdin.readline

N, M = map(int, input().split())
members = list(map(int, input().split()))

# 최고 판단 리스트 (1이면 최고)
cnt_list = [1] * N

for i in range(M):
    # f, s : 비교할 인덱스 + 1
    f, s = map(int, input().split())

    if members[f-1] < members[s-1]:
        cnt_list[f-1] = 0
    elif members[f-1] > members[s-1]:
        cnt_list[s-1] = 0
    elif members[f-1] == members[s-1]:
        cnt_list[f-1] = 0
        cnt_list[s-1] = 0

print(cnt_list.count(1))