본문 바로가기
PS/AtCoder

[ABC301] A - Overall Winner (AtCoder Beginner Contest 301 in Java)

by Nahwasa 2023. 5. 13.

문제 : abc301 - A

 

필요 알고리즘

  • 구현, 문자열
    • 문제에서 제시된 대로 구현하는 문제이다.

※ 제 코드에서 왜 main 함수에 로직을 직접 작성하지 않았는지, 왜 Scanner를 쓰지 않고 BufferedReader를 사용했는지 등에 대해서는 '자바로 백준 풀 때의 팁 및 주의점' 글을 참고해주세요. 백준을 자바로 풀어보려고 시작하시는 분이나, 백준에서 자바로 풀 때의 팁을 원하시는 분들도 보시는걸 추천드립니다.

 

 

풀이

  문자열의 각 문자를 읽는방법을 안다면 풀 수 있다.

T와 A가 나온 갯수를 세서 더 많은 쪽의 문자를 출력해주면 된다.

다만 갯수가 동일하다면 최대치에 더 먼저 도달한 문자를 출력해줘야 한다. 이 부분은 코드에서 'c' 변수로 처리했다.

 

 

코드 : github

import java.io.BufferedReader;
import java.io.InputStreamReader;
 
public class Main {
 
    static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
 
    public static void main(String[] args) throws Exception {
        new Main().solution();
    }
 
    private void solution() throws Exception {
        br.readLine();
        String s = br.readLine();
        int cnt1 = 0;
        int cnt2 = 0;
        int max = 0;
        char c = ' ';
 
        for (int i = 0; i < s.length(); i++) {
            if (s.charAt(i) == 'A') cnt1++;
            else cnt2++;
            
            if (cnt1 > max) {
                max = cnt1;
                c = 'A';
            }
            if (cnt2 > max) {
                max = cnt2;
                c = 'T';
            }
        }
 
        System.out.println(cnt1==cnt2 ? c : (cnt1>cnt2 ? 'A' : 'T'));
    }
}

 

댓글