Level 2. 롤케이크 자르기

Level 2. 롤케이크 자르기

 

 

토핑의 번호가 특별한 의미를 갖지 않기 때문에 map 대신 unordered_map을 사용했다.

• C++의 맵 자료구조는 한 번 접근한 키는 값이 0이어도 size에 포함되기 때문에, erase를 해주어야 한다.

 

#include <iostream>
#include <vector>
#include <unordered_map>

using namespace std;

int solution(vector<int> topping)
{
    int answer = 0;
    unordered_map<int, int> umapA, umapB;

    for (int tp : topping)
        ++umapB[tp];

    for (int tp : topping)
    {
        ++umapA[tp];

        if (--umapB[tp] <= 0)
            umapB.erase(tp);

        if (umapA.size() == umapB.size())
            ++answer;
    }

    return answer;
}