프로그래머스/lv.2

롤케이크 자르기

bingual 2024. 2. 20. 15:23
반응형

 

https://school.programmers.co.kr/learn/courses/30/lessons/132265

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

문제풀이

슬라이스 + 집합을 반복문 내에서 사용하게 된다면 O(n^3) 시간복잡도가 발생할 수 있다.

 

  • 각 횟수를 기록한 a, 집합 b를 선언
  • a는 등장한 횟수 만큼 감소 시켜 0이 되면 제외, b는 등장한 횟수 만큼 추가
  • 각 길이가 같을 때 카운트
from collections import Counter


def solution(topping):
    answer = 0

    a = Counter(topping)
    b = set()
    for num in topping:
        a[num] -= 1
        if a[num] <= 0:
            del a[num]

        b.add(num)

        if len(a) == len(b):
            answer += 1

    return answer