자바(Java) 알고리즘 문제풀이 학급 회장 [인프런 / Hash]

출처) 인프런

강의의 문제를 가져왔기 때문에 밝힙니다 ! 
임의로 추가, 수정, 삭제한 내용들이 있으니 정확한 이해를 위해서는 강의를 구매하시는 것을 추천드립니다 😅
출처 : 자바(Java) 알고리즘 문제풀이 : 코딩테스트 대비 (인프런 강의)

자바(Java) 알고리즘 문제풀이 : 코딩테스트 대비 - 인프런 | 강의

 

1) 강의 보기 전의 풀이

HashMap 을 사용해서 풀었는데, 학급 회장 선거에 나오는 학생 수가 5로 고정되어 있어서 put() 을 5번 사용해 풀이했다.

이러면 선거에 나오는 인원 수가 다섯 명이 아닐 때는 풀이할 수 없어서 고민을 해봤는데, 다른 대안이 생각나지 않아서 맞췄긴 했어도

강의를 한번 시청.. 😅

package study.section4;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.HashMap;
import java.util.Map;

public class Section0401 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int studentCount = Integer.parseInt(br.readLine());
        String line = br.readLine();
        String[] votes = line.split("");

        String answer = solution(studentCount, votes);
        System.out.println(answer);
    }

    private static String solution(int studentCount, String[] votes) {
        String answer = "";

        // TODO <후보 key, 몇 명이 뽑았는지 value>
        Map<String, Integer> resultMap = new HashMap<>();
        resultMap.put("A", 0);
        resultMap.put("B", 0);
        resultMap.put("C", 0);
        resultMap.put("D", 0);
        resultMap.put("E", 0);

        for (String key : resultMap.keySet()) {
            for (int i = 0; i < studentCount; i++) {
                // TODO key 와 vote[i] 가 일치하면 해당 알파벳에 카운트를 +1 시킨다.
                if (key.equals(votes[i])) {
                    resultMap.put(key, resultMap.get(key) + 1);
                }
            }
        }

        // TODO 반복문이 끝나면 resultMap 에 데이터가 담긴다.
        //  담긴 데이터를 가지고 최대 값 알고리즘 적용
        int max = 0; // resultMap 의 value

        for (int i = 0; i < resultMap.size(); i++) {
            for (String key : resultMap.keySet()) {
                Integer integer = resultMap.get(key);

                if (integer > max) {
                    max = integer;
                    answer = key;
                }
            }
        }

        return answer;
    }
}

// 15
// B A C B A C C A C C A C C B D E D E
// C 당선

 

2. 강의에서 얻은 힌트

강의를 보고 HashMap 에서 제공하는 getOrDefault() 메서드에 대해 알게 되었다.

해당 메서드를 사용하면, map 에 Key 값이 존재하면 그 Key 의 Value 에 +1 해서 갱신이 되고, Key 값이 존재하지 않다면 Value 가 1인

새로운 Key 가 추가된다 !

이렇게 간단하게 로직 길이도 줄이고, 몇 명이 선거에 나오든 풀이할 수 있게 되었다.

private static String solution(int studentCount, String[] votes) {
    String answer = " ";

    Map<String, Integer> map = new HashMap<>();

    for (String str : votes) {
        // get() 은 key 의 value 값을 가져오는데, str 라는 key 가 map 에 없을 수도 있으니까 
        // 이땐 갱신이 아닌 추가를 위해 getOrDefault() 사용
        //  key   - A, B, C, D
        //  value - 1, 3, 3, 1
        //  이때 만약 str 이 E 라면, E 는 없으니까 map 에 <E, 0+1> 추가
        //  이떄 만약 str 이 A 라면, A 는 있으니까 mpa 에 <A, 1+1> 갱신
        map.put(str, map.getOrDefault(str, 0) + 1);
    }
    ... 중략 ...
}