문제 바로가기

나의 풀이

투포인터나 슬라이딩 윈도우를 활용하면 되는 문제이다. 사실 성능을 더 높이기 위해서는 remove 했을때 마이너스 값이 된 것이 있으면 add와 세트로 모든 값이 다시 0이상이 될때까지 반복되는 loop를 만들어주면 된다.

다만 여기서는 그냥 단순하게 1칸씩만 이동하면서 매번 check 해주는 것으로 하였다.

나의 코드

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

class Main {

    static int answer;
    static String str;
    static Map<Character, Integer> require;

    public static void main(String[] args) throws IOException {

        answer = 0;
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int entire = Integer.parseInt(st.nextToken());
        int part = Integer.parseInt(st.nextToken());
        str = br.readLine();

        require = new HashMap<>();
        st = new StringTokenizer(br.readLine());
        require.put('A', -Integer.parseInt(st.nextToken()));
        require.put('C', -Integer.parseInt(st.nextToken()));
        require.put('G', -Integer.parseInt(st.nextToken()));
        require.put('T', -Integer.parseInt(st.nextToken()));

        addInRange(0, part);

        int left = 0;
        int right = part;
        while (true) {
            check();
            if (right >= entire) {
                break;
            }
            removeInRange(left, left + 1);
            addInRange(right, right + 1);
            left++;
            right++;
        }

        System.out.println(answer);
    }

    private static void check() {
        for (Integer value : require.values()) {
            if (value < 0) {
                return;
            }
        }
        answer++;
    }

    private static void addInRange(int start, int end) {
        for (int i = start; i < end; i++) {
            require.compute(str.charAt(i), (character, integer) -> integer + 1);
        }
    }

    private static void removeInRange(int start, int end) {
        for (int i = start; i < end; i++) {
            require.compute(str.charAt(i), (character, integer) -> integer - 1);
        }
    }

}