백준: 17299 오등큰수

문제

17299번: 오등큰수

 

문제 풀이

 바로 앞 번 문제인 17298 오큰수를 이해하였다면 쉽게 풀 수 있는 문제이다. 앞의 문제는 단순히 수열의 값을 비교하여 큰 값이 있는지 확인하는 문제였지만, 이 문제는 각 수열과 중복되는 값을 카운트하여 카운트 되는 값보다 큰 값이 있는지 확인하는 문제이다.😉

 

코드

from sys import stdin
from collections import Counter

if __name__ == "__main__":
    stack = []

    n = int(stdin.readline())
    s = list(stdin.readline().split())
    cnt = Counter(s)
    s = [[cnt[num], int(num)] for num in s]
    answer = [-1 for _ in range(n)]
    stack.append(0)

    i = 1

    while stack and i < n:
        while stack and s[stack[-1]][0] < s[i][0]:
            answer[stack[-1]] = s[i][1]
            stack.pop()

        stack.append(i)
        i += 1

    for num in answer:
        print(num, end=' ')

 오큰수 문제와 코드는 거의 동일하며, 각 원소의 개수를 카운트 하기 위해 Counter를 추가하고, 카운터 값을 기준으로 대소를 비교하는 것 말고는 차이가 없다.