본문 바로가기
프로그래머스/lv.2

롤케이크 자르기

by bingual 2024. 2. 20.
반응형

 

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

'프로그래머스 > lv.2' 카테고리의 다른 글

[3차] 파일명 정렬  (0) 2024.02.20
오픈채팅방  (0) 2024.02.20
스킬트리  (0) 2024.02.20
땅따먹기  (0) 2024.02.18
주식가격  (0) 2024.02.17