[프로그래머스][JAVA]Lv. 2 - 롤케이크 자르기 문제

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

프로그래머스

동생과 두 조각을 나눠 먹는데, 토핑의 개수가 같아야 한다.

롤케이크의 전체 토핑 종류의 개수를 HashMap `map` 의 size로 판단하기로 하고,

index = 0부터 `map2` 에 토핑 종류의 개수를 추가해나가고, `map` 에서는 제거해나가서

`map` 과 `map2` 의 size 가 동일할 때, answer 를 증가시켜서 문제를 해결하였다.

 

import java.util.HashMap;
class Solution {
    public int solution(int[] topping) {
        int answer = 0;

        HashMap<Integer, Integer> map = new HashMap<>();
        HashMap<Integer, Integer> map2 = new HashMap<>();

        for (int k : topping) {
            if (!map.containsKey(k)) {
                map.put(k, 1);
            } else {
                map.put(k, map.get(k) + 1);
            }
        }
        
        for (int j : topping) {
            map.put(j, map.get(j) - 1);
            if (!map2.containsKey(j)) {
                map2.put(j, 1);
            } else {
                map2.put(j, map2.get(j) + 1);
            }
            if (map.containsKey(j) && map.get(j) <= 0) {
                map.remove(j);
            }
            if (map.size() == map2.size()) answer++;
        }
        
        return answer;
    }
}

HashMap 은 이미 존재하는 키면 중복되서 넣어지지 않는 것을 이용

점점 줄어드는 `map` 은 키가 존재하면서 그 값이 0 보다 작아진다면 제거하는 방식으로 size를 조절했다.

예시 [1, 2, 1, 3, 1, 4, 1, 2]에서 정답인 경우에 map을 출력해 본다면,

map = {1=2, 2=1, 4=1}
map2 = {1=2, 2=1, 3=1}

map = {1=1, 2=1, 4=1}
map2 = {1=3, 2=1, 3=1}이다.

이 경우에서 두 롤케이크 조각의 토핑의 갯수가 같다.