자바 알고리즘 문제 풀이 입문. 섹션4. HashMap, TreeSet (해쉬, 정렬지원 Set) : 학급 회장

 

https://hyejin.tistory.com/1226

자바 알고리즘 문제 풀이 입문. 섹션3. Two Pointers, Sliding Window [효율성 : O(n^2) --> O(n)] : 최대 길이

-> 이전 문제 풀이 

 

 

1. 학급 회장 

설명

학급 회장을 뽑는데 후보로 기호 A, B, C, D, E 후보가 등록을 했습니다.

투표용지에는 반 학생들이 자기가 선택한 후보의 기호(알파벳)가 쓰여져 있으며 선생님은 그 기호를 발표하고 있습니다.

선생님의 발표가 끝난 후 어떤 기호의 후보가 학급 회장이 되었는지 출력하는 프로그램을 작성하세요.

반드시 한 명의 학급회장이 선출되도록 투표결과가 나왔다고 가정합니다.

 

입력

첫 줄에는 반 학생수 N(5<=N<=50)이 주어집니다.

두 번째 줄에 N개의 투표용지에 쓰여져 있던 각 후보의 기호가 선생님이 발표한 순서대로 문자열로 입력됩니다.

 

출력

학급 회장으로 선택된 기호를 출력합니다.

 

예시 입력 1 

15
BACBACCACCBDEDE

 

예시 출력 1

C

 

 

문제 풀이 1

public String solution(int n, String result) {
    String answer = "";
    Map<String, Integer> map = new HashMap<>();
    String[] split = result.split("");
    for (String s : split) {
        Integer sum = map.getOrDefault(s, 1);
        map.put(s, sum + 1);
    }

    int max = Integer.MIN_VALUE;
    for (String s : map.keySet()) {
        Integer score = map.get(s);
        if (max < score) {
            answer = s;
            max = score;
        }
    }

    return answer;
}

👩🏻‍💻 : 우선 이번 섹션부터는 HashMap, TreeSet 을 사용해서 문제 푸는 부분이다. 

먼저 문제에서 후보는 A, B, C, D, E 5명이라고 한다. 그러므로 A, B, C, D, E 가 HashMap의 key 가 된다. 

그리고 주어진 문자열 result 에서 공백을 기준으로 split하고, 반복문을 돌려서 map에 해당 키에 score 을 더해줘야 하는데 

map의 맨 처음에는 아무 값도 없기 때문에 getOrDefault 메서드를 사용해서 key 값이 있다면 그 값을 구해서 +1 해주고, 없다면 0 을 반환한 다음 + 1 을 해준다. 

 

이렇게 키 값에 해당하는 투표 결과를 모두 저장한 다음에 

map.keySet 메서드를  통해서 키 값으로 반복문을 돌려 map에서 이 키값을 get 한 값이 max 값보다 크면 answer 에 해당 키 값을 저장해주고, max 값은 해당 키 값으로 대입해준다. 

 

 

문제 풀이 2

public char solution2(int n, String s) {
    char answer = ' ';
    HashMap<Character, Integer> map = new HashMap<>();
    for (char x : s.toCharArray()) {
        map.put(x, map.getOrDefault(x, 0) + 1);
    }

    int max = Integer.MIN_VALUE;
    for (char key : map.keySet()) {
        if (map.get(key) > max) {
            max = map.get(key);
            answer = key;
        }
    }

    return answer;
}

👾 : 강사님이 푼 방식은 나와 동일한데 차이점이라고 한다면 문자열을 split이 아닌 Character 형식으로 바꿨다는 것이다.