백준 알고리즘 | 1157번 문제: 단어 공부

출처

https://www.acmicpc.net/problem/1157

문제

알파벳 대소문자로 된 단어가 주어지면, 이 단어에서 가장 많이 사용된 알파벳이 무엇인지 알아내는 프로그램을 작성하시오. 단, 대문자와 소문자를 구분하지 않는다.

입력

첫째 줄에 알파벳 대소문자로 이루어진 단어가 주어진다. 주어지는 단어의 길이는 1,000,000을 넘지 않는다.

출력

첫째 줄에 이 단어에서 가장 많이 사용된 알파벳을 대문자로 출력한다. 단, 가장 많이 사용된 알파벳이 여러 개 존재하는 경우에는 ?를 출력한다.

예제 입력 1

Mississipi

예제 출력 1

?

예제 입력 2

zZa

예제 출력 2

Z

예제 입력 3

baaa

예제 출력 3

A

알고리즘 분류

  • 수학
  • 구현

문제 풀이

통과한 코드 (1)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class Main {

    public static void main(String[] args) throws IOException {
        try (BufferedReader br = new BufferedReader(new InputStreamReader(System.in))) {
            // 입력
            String input = br.readLine();

            Map<Character, Integer> alphabetCountMap = new HashMap<>();

            Set<Character> maxSet = new HashSet<>();
            int max = Integer.MIN_VALUE;

            for (int i = 0; i < input.length(); i++) {
                Character alphabet = Character.toUpperCase(input.charAt(i));

                int count;
                if (alphabetCountMap.containsKey(alphabet)) {
                    count = alphabetCountMap.get(alphabet) + 1;
                } else {
                    count = 0;
                }
                alphabetCountMap.put(alphabet, count);

                if (max < count) {
                    max = count;
                    maxSet.clear();
                    maxSet.add(alphabet);
                } else if (max == count) {
                    maxSet.add(alphabet);
                }
            }

            // 출력
            if (maxSet.size() > 1) {
                System.out.println("?");
            } else {
                System.out.println(maxSet.iterator().next());
            }
        }
    }
}

테스트 코드가 통과를 하긴 했는데, 속도 504ms으로 너무 느리다.

통과한 코드 (2)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main2 {

    public static void main(String[] args) throws IOException {
        try (BufferedReader br = new BufferedReader(new InputStreamReader(System.in))) {
            // 입력
            String input = br.readLine();

            // 알파벳 수를 카운트한다.
            int[] alphabet = new int[26];
            for (int i = 0; i < input.length(); i++) {
                //char c = Character.toUpperCase(input.charAt(i));
                //alphabet[c - 65]++;
                char c = input.charAt(i);
                if (c >= 'a')
                    alphabet[c - 'a']++;
                else
                    alphabet[c - 'A']++;
            }

            // 많이 나온 알바벳을 구한다.
            char output = 0;
            int max = Integer.MIN_VALUE;
            for (int i = 0; i < alphabet.length; i++) {

                if(max < alphabet[i]) {
                    max = alphabet[i];
                    output = (char) (i + 'A');
                } else if(max == alphabet[i]) {
                    output = '?';
                }
            }

            // 출력
            System.out.println(output);
        }
    }
}

속도가 264ms으로 개선되었다.