[자바][Java][백준][1157번] 단어 공부 - 탐색

문제 설명

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

입력

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

출력

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

 

생각

처음엔 HashMap을 생각하여 푸려고 했다. 각 문자를 탐색하면서 HashMap에 이미 있으면 1을 더하고 아니면 새롭게 1을 넣는 방식을 생각했다. 근데 왜 70퍼에서 실패가 뜨는지 모르겠다..

 

몇 번의 실패를 하고나서, 배열로 해결하자고 생각했다. 알파벳이고 대문자나 소문자로 통일해야하므로 크기가 26일 배열을 사용하면 충분할 것이다.

 

대문자로 바꾸고 char형태로 바꾼 후 'A' 씩 빼주면 0~25 사이의 숫자가 될 것이다. 후에는 각 인덱스에 맞게 1씩 추가해주면 될 것이다. 자세한 것은 주석에 적겠다.

 

코드

 

import java.util.Scanner;
 
public class Main {
 
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String str = sc.next().toUpperCase();
        int max = Integer.MIN_VALUE;
        int index = -1;
        int count = 0;
        
        int[] arr = new int[26];
        
        for(int i=0; i<str.length(); i++) {
            char tmp = str.charAt(i);
            //'A'를 빼줘야 인덱스가 0~25가 될 수 있다.
            arr[tmp-'A']++;
        }
        
        //배열을 돌면서 max를 찾아준다.
        for(int i=0; i<arr.length; i++) {
            if(max < arr[i]) {
                max = arr[i];
            }
        }
        
        //max가 여러개 있는지 확인한다.
        //또한 1개일 경우에는 그 인덱스를 저장하는 변수를 사용한다.
        for(int i=0; i<arr.length; i++) {
            if(max == arr[i]) {
                count++;
                index = i;
            }
        }
        
        //출력
        if(count > 1) {
            System.out.println("?");
        } else {
            System.out.println((char) (index + 'A'));
        }
    }
 
}