[java]2022 KAKAO BLIND RECRUITMENT> 신고 결과 받기, 배열에서 특정 요소 값 index 찾기, map.getOrDefault(,)

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

프로그래머스

 

코드

 

import java.util.*;
class Solution {
    public int[] solution(String[] id_list, String[] report, int k) {
      int[] answer = new int[id_list.length];
		boolean[][] check = new boolean[id_list.length][id_list.length];
		Map map = new HashMap();
		for (int i = 0; i < report.length; i++) {// 신고한 사람 찾아주고 신고한 횟수 구하기
			String[] ur = report[i].split(" ");

			int a = Arrays.asList(id_list).indexOf(ur[0]); // 신고한 사람 인덱스 찾기
			int b = Arrays.asList(id_list).indexOf(ur[1]);
			if (check[a][b] != true) {
				check[a][b] = true;
				if (map.containsKey(ur[1])) {
					// 키가 이미 존재한다면
					Integer value = (Integer) map.get(ur[1]);
					map.put(ur[1], new Integer(value.intValue() + 1));
				} else {// 키가 존재하지 않는 경우
					map.put(ur[1], new Integer(1));
				}
			}

		}

		for (int i = 0; i < id_list.length; i++) {
			int cnt = 0;
			for (int j = 0; j < id_list.length; j++) {

				if (check[i][j] == true && (Integer) map.getOrDefault(id_list[j], 0) >= k) {
					cnt++;
				}
			}
			answer[i] = cnt;
		}

		return answer;
    }
}

 

해설

1. boolean[][] check이차원 배열을 만들어 신고를 한 사람을 체크하여 중복을 막아줌
2. 신고를 당한 사람이 과거에 같은 사람에게 안 받았을 경우에 횟수 ++
3. check를 확인하며 신고한 사람이 정지 횟수 넘었을 때 메일 횟수++

 

배열에서 특정 요소 값 index 찾기
Arrays.asList(id_list).indexOf(ur[0]);

map에서 찾은 키 값이 존재하지 않으면 defalut값 지정

map.getOrDefault(id_list[j], 0)