[Programmers] 롤케이크 자르기 (JAVA)

프로그래머스

 

풀이

 

투포인터처럼 반복문 1번으로 문제를 해결하려고 노력했다.

HashMap 2개를 사용해서, map의 size가 같으면 토핑의 종류가 같은 거라 answer++

반복문을 돌면서 철수가 가진 토핑을 빼주고, 동생에게 더해주며 map의 size를 비교해줬다.

 

전체코드

import java.io.*;
import java.util.*;

class Solution {
    public int solution(int[] topping) {
        HashMap<Integer,Integer> chulsu = new HashMap<>();
        HashMap<Integer,Integer> sister = new HashMap<>();
        
        for(int i=0; i<topping.length-1; i++){
            if(!chulsu.containsKey(topping[i])){
                chulsu.put(topping[i], 1);
            }else{
                chulsu.put(topping[i], chulsu.get(topping[i])+1);
            }
        }
        
        sister.put(topping[topping.length-1], 1);
                
        int answer = 0;
        
        if(chulsu.size() == sister.size()){
            answer++;
        }
        
        for(int i=topping.length-2; i>=0; i--){            
            if(chulsu.get(topping[i]) == 1){
                chulsu.remove(topping[i]);
            }else{
                chulsu.put(topping[i], chulsu.get(topping[i])-1);
            }
            
            if(sister.containsKey(topping[i])){
                sister.put(topping[i], sister.get(topping[i])+1);
            }else{
                sister.put(topping[i], 1);
            }
            
            if(chulsu.size() == sister.size()){
                answer++;
            }
        }

        return answer;
    }
}