[BOJ/Java] 12891. DNA 비밀번호

[BOJ/Java] 12891. DNA 비밀번호

12891번: DNA 비밀번호

---

풀이

package edu.ssafy.im.BOJ.S2.No12891;

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

public class Main {
    static int s, p, ans = 0; // DNA 문자열의 길이, 부분문자열의 길이, 비밀번호로 사용할 수 있는 종류의 수
    static int arr[]; // DNA 문자열을 저장하는 배열
    static int c[] = new int[4]; // 부분문자열에 포함되어야 할 각 문자의 최소 개수
    static int r[] = new int[4]; // 부분문자열의 각 문자의 개수를 저장하는 배열

    public static void main(String[] args) throws NumberFormatException, IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        String input = br.readLine();
        StringTokenizer st = new StringTokenizer(input);
        s = Integer.parseInt(st.nextToken());
        p = Integer.parseInt(st.nextToken());

        input = br.readLine();
        arr = new int[s];
        for (int i = 0; i < s; i++) {
            char c = input.charAt(i);
            // 문자에 따라 숫자로 변환하여 배열에 저장
            switch (c) {
                case 'A':
                    arr[i] = 0;
                    break;
                case 'C':
                    arr[i] = 1;
                    break;
                case 'G':
                    arr[i] = 2;
                    break;
                case 'T':
                    arr[i] = 3;
                    break;
            }
        }

        input = br.readLine();
        st = new StringTokenizer(input);
        for (int i = 0; i < 4; i++) {
            // 각 문자의 최소 개수 입력 받기
            c[i] = Integer.parseInt(st.nextToken());
        }

        // 첫 번째 부분문자열에 대한 각 문자의 개수 계산
        for (int i = 0; i < p; i++) {
            r[arr[i]]++;
        }

        int ans = 0;
        int start = 0, end = 0 + p - 1; // 부분문자열의 시작과 끝 인덱스 설정
        // 모든 가능한 부분문자열에 대해 반복
        for (int i = 0; i < s - p + 1; i++) {
            // 부분문자열에 대해 각 문자의 개수가 최소 개수보다 큰지 확인
            for (int j = 0; j < 4; j++) {
                if (r[j] < c[j]) {
                    break;
                } else if (j == 3) {
                    ans++; // 모든 문자의 개수가 최소 개수 이상인 경우, 비밀번호로 사용 가능한 경우임을 카운트
                }
            }
            try {
                // 다음 부분문자열의 시작과 끝 인덱스를 설정하고 각 문자의 개수 업데이트
                r[arr[start++]]--;
                r[arr[++end]]++;
            } catch (Exception e) {
            }
        }
        sb.append(ans); // 결과 문자열에 비밀번호로 사용 가능한 부분문자열의 개수 추가
        System.out.println(sb); // 결과 출력
    }
}