[백준] 12891번: DNA 비밀번호

책: Do It 알고리즘 코딩테스트 Python편

 

문제:

009 DNA 비밀번호

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

12891번: DNA 비밀번호

 

코드:

check = [0] * 4

def remove(c):
    if c == 'A':
        check[0] -= 1 
    elif c == 'C':
        check[1] -= 1
    elif c == 'G':
        check[2] -= 1
    else:
        check[3] -= 1
        
def add(c):
    if c == 'A':
        check[0] += 1
    elif c == 'C':
        check[1] += 1
    elif c == 'G':
        check[2] += 1
    else:
        check[3] += 1

S, P = map(int, input().split())
L = str(input())
AC, CC, GC, TC = map(int, input().split())
cnt = 0

for i in range(len(L)):
    add(L[i])
    #print(str(check[0]) + ',' + str(check[1]) + ',' +str(check[2]) + ',' +str(check[3]))

    if i + 1 -(P-1) > 0:
        if check[0] >= AC and check[1] >= CC and check[2] >= GC and check[3] >= TC:
            cnt += 1
        remove(L[i + 1 - P])

print(cnt)