[백준] 12891 DNA 비밀번호 - 투포인터(Two Pointer) JAVA

https://www.acmicpc.net/problem/12891

코드설명

투포인터(Two Pointer)를 활용합니다.

 

문제에서 P와 S의 크기는 10^6 입니다. 

만약 P^2 과 같은 제곱수의 연산이 일어난다면, 10^12 으로써 , 엄청나게 많은 시간이 걸리므로 이 문제는 선형시간(O(N))에 풀어내야한다는 것을 유추할 수 있습니다.

 

문제를 보면 부분문자열의 길이 P가 주어집니다. 그러므로, 투포인터를 사용한다는 생각을 할 수 있습니다.

코드

정답코드입니다.

package Main;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.StringTokenizer;
 
public class Main {
	static int N, S, P;
	static int answer = 0;
	static char[] arr;
	static int[] cnt;
	public static void main(String[] args) throws IOException{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		
		S = Integer.parseInt(st.nextToken());
		P = Integer.parseInt(st.nextToken());
		
		st = new StringTokenizer(br.readLine());
		arr = new char[S];
		arr = st.nextToken().toCharArray();

		cnt = new int[4];
		st = new StringTokenizer(br.readLine());
		for(int i=0;i<4;i++) {
			cnt[i] = Integer.parseInt(st.nextToken());
		}
		
		twoPointer();
		System.out.println(answer);
	}
	
	static void twoPointer() {
		
		int[] cntStr = new int[4];
		for(int i=0;i<P-1;i++) {
			if(arr[i] == 'A') {
				cntStr[0] += 1;
			}else if(arr[i] == 'C') {
				cntStr[1] += 1;
			}else if(arr[i] == 'G') {
				cntStr[2] += 1;
			}else if(arr[i] == 'T') {
				cntStr[3] += 1;
			}
		}
		
		for(int i=P-1; i<S; i++) {
			
			if(arr[i] == 'A') {
				cntStr[0] += 1;
			}else if(arr[i] == 'C') {
				cntStr[1] += 1;
			}else if(arr[i] == 'G') {
				cntStr[2] += 1;
			}else if(arr[i] == 'T') {
				cntStr[3] += 1;
			}
			
			boolean successFlag = true;
			for(int j=0; j<4; j++) {
				if(cnt[j] > cntStr[j]) {
					successFlag = false;
					break;
				}
			}
			
			if(successFlag == true) {
				answer += 1;
			}
			
			if(arr[i-(P-1)] == 'A') {
				cntStr[0] -= 1;
			}else if(arr[i-(P-1)] == 'C') {
				cntStr[1] -= 1;
			}else if(arr[i-(P-1)] == 'G') {
				cntStr[2] -= 1;
			}else if(arr[i-(P-1)] == 'T') {
				cntStr[3] -= 1;
			}
			
		}
		
	}
	
}