[프로그래머스] 신고 결과 받기

nayoon·2022년 3월 10일

코딩테스트

목록 보기

1/1

오랜만에 풀어도 이렇게 오래 걸릴 문제는 아니었는데, 개인적으로는 많이 아쉽다.

(작년 카카오 코테에서는 깔끔은 아니어도 빨리 풀었던 거 같은데.. 이래서 시간 압박이 중요한가보다..)

내 풀이

import java.util.HashMap;
import java.util.Map;
import java.util.StringTokenizer;
import java.util.Arrays;

public class Programmers_신고_결과_받기 {
	
    public static int[] solution(String[] id_list, String[] report, int k) {
        int[] answer = new int[id_list.length];
        Map<String, Map<String, Boolean>> map = new HashMap<>();
        Map<String, Integer> success = new HashMap<>();
        
        for(String id: id_list) {
        	map.put(id, new HashMap<>());
        	success.put(id, 0);
        }
        StringTokenizer st;
        String reporter, attacker;
        Map<String, Boolean> temp;
        
        for(String rep: report) {
        	st = new StringTokenizer(rep);
        	reporter = st.nextToken();
        	attacker = st.nextToken();
        	
        	temp = map.get(reporter);
        	if (!temp.getOrDefault(attacker, false)) {
	        	temp.put(attacker, true);
	        	
	        	success.put(attacker, success.get(attacker) + 1);
        	}
        	
        }
        
        for(String id: id_list) {
        	if (success.get(id) >= k) {
        		for(int i = 0; i < id_list.length; i++) {
        			if (map.get(id_list[i]).containsKey(id)) answer[i]++;
        		}
        	}
        }
        
        return answer;
    }	
    
	public static void main(String[] args) {
		System.out.println(Arrays.toString(solution(new String[] {"muzi", "frodo", "apeach", "neo"}, new String[] {"muzi frodo","apeach frodo","frodo neo","muzi neo","apeach muzi"}, 2)).toString());
		System.out.println(Arrays.toString(solution(new String[] {"con", "ryan"}, new String[] {"ryan con", "ryan con", "ryan con", "ryan con"}, 3)).toString());
	}
}