[BackJoon]1157번 스트림 사용해서 풀기

문제

 

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

단, 대문자와 소문자를 구분하지 않는다.

 

입력

 

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

 

출력

 

첫째 줄에 이 단어에서 가장 많이 사용된 알파벳을 대문자로 출력한다. 

단, 가장 많이 사용된 알파벳이 여러 개 존재하는 경우에는 ?를 출력한다.

 

 

풀이

 

문자열을 활용한 비슷비슷한 문제다.

 

알파벳의 개수인 26개의 원소를 가진 리스트를 하나 생성해 모두 0으로 채운다.

 

계속해서 주어진 문자열을 알파벳으로 분해 후 모든 문자를 대문자로 변경, -65를 해서 A = 0으로 맞춘다.

 

이어서 문자열의 길이만큼 for문을 돌며 해당 알파벳의 등장시마다 리스트의 원소 값을 +1 해준다.

 

그다음 스트림을 이용해 리스트를 내림차순으로 정렬,

 

첫 번째 원소와 두 번째 원소의 값이 같은지 확인해서(같다면 많이 등장하는 알파벳이 두 개 이상이라는 뜻이므로)

 

같다면 "?"를, 아니면 해당 횟수만큼 나온 알파벳을 다시 char로 변경해 출력한다.

 

import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.Scanner;
import java.util.stream.Collectors;

public class Prob1157 {
    public static void main(String[] args) {

        Scanner sc = new Scanner(System.in);

        List<Integer> result = new ArrayList<>();

        for (int i = 0; i < 26; i++) {
            result.add(0);
        }

        List<Integer> ch = sc.nextLine().toUpperCase().chars()
                .map(a -> a - 65)
                .boxed()
                .collect(Collectors.toList());

        for (int i = 0; i < ch.size(); i++) {
            result.set(ch.get(i), result.get(ch.get(i)) + 1);
        }

        List<Integer> sortedResult = result.stream()
                .sorted(Comparator.reverseOrder())
                .collect(Collectors.toList());

        if (sortedResult.get(0).equals(sortedResult.get(1))) System.out.println("?");
        else System.out.println((char) (result.indexOf(sortedResult.get(0)) + 65));
    }
}