[알고리즘]학급 회장구하기

▷ 문제

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

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

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

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

* 입력

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

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

* 출력

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

▷ 입력 예시

15

BACBACCACCBDEDE

▷ 출력 예시

C

▷ 풀이

import java.util.Scanner;
import java.util.HashMap;

public class Main {	
    public char solution(int n, String str){
      char answer = ' ';
      HashMap<Character, Integer> map = new HashMap<>();
      int max = Integer.MIN_VALUE;

      for(char x : str.toCharArray()){
        map.put(x, map.getOrDefault(x,0) + 1);
      }

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

      return answer;       
    }

    public static void main(String[] args){
        Main main = new Main();
        Scanner kb = new Scanner(System.in);

        int n = kb.nextInt();
        String str = kb.next();

        System.out.println(main.solution(n, str));
    }
  }

▷ 핵심 포인트

1. map.get(x)에 값이 없을 수도 있으므로 map.getOrDefault를 사용하여 값이 존재하지 않을 경우의 값을 설정해줍니다.