[Programmers] 롤케이크 자르기 - 132265

from collections import Counter 사용.

전체 list 두 개를 set해서 len() 비교를 하면 시간초과가 발생하기 때문에 Counter를 이용한다 !

 

Question

 

롤케이크에 올려진 토핑들의 번호를 저장한 정수 배열 topping이 매개변수로 주어질 때, 롤케이크를 공평하게 자르는 방법의 수를 return 하도록 solution 함수를 완성해주세요.

Point

• for문을 돌리며 두 개로 나눈 list를 전체 set 해서 len() 비교는 시간 초과됨. => 따라서 실패.
• Counter를 이용하여 한 명(p1)에게 전부 몰아 넣었다고 가정하고 for문을 통해 하나씩 다른 사람(p2)에 전달하고 len() 비교.

 

Code

 

from collections import Counter
def solution(topping):
    answer = 0
    p1 = Counter(topping)
    p2 = set()
    for i in topping:
        p1[i] -= 1
        if p1[i] == 0 : del p1[i]
        p2.add(i)
        if len(p1) == len(p2) : answer += 1
    
    return answer

 

now me

On my github