본문 바로가기
PS/BOJ

[자바] 백준 25841 - Digit Count (java)

by Nahwasa 2023. 1. 19.

 문제 : boj25841


 

필요 알고리즘 개념

  • 문자열, 구현
    • 문자열을 적절히 다뤄서 주어진 조건대로 구현해주면 된다.

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

 


 

풀이

  최악의 경우 1000부터 9999까지이고, 각 숫자는 4자리밖에 안된다. 그러므로 1000부터 9999까지 모든 숫자를 보면서 모든 자리수에 대해 확인해봐도 O(9000*4) 면 된다!

 

1. 시작숫자와 끝 숫자를 입력받고, 카운팅하려는 문자를 받는다.

2. 시작숫자부터 끝 숫자까지 반복문을 돌리면서, 해당 숫자를 문자로 변경해준다.

3. 변경한 문자의 각 character가 '1'에서 입력받은 카운팅하려는 문자와 동일하면 카운팅해준다.

 


 

코드 : github

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
    private void solution() throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in), 1<<16);
        StringTokenizer st = new StringTokenizer(br.readLine());
        int a = Integer.parseInt(st.nextToken());
        int b = Integer.parseInt(st.nextToken());
        char c = st.nextToken().charAt(0);
        int cnt = 0;
        for (int i = a; i <= b; i++) {
            String cur = String.valueOf(i);
            for (int j = 0; j < cur.length(); j++)
                if (cur.charAt(j) == c) cnt++;
        }
        System.out.println(cnt);
    }

    public static void main(String[] args) throws Exception {
        new Main().solution();
    }
}

댓글