Lesson 5: Prefix Sums → Genomic Range Query

문제

https://app.codility.com/programmers/lessons/5-prefix_sums/genomic_range_query/
DNS 시퀀스는 A, C, G, T로 나타낼 수 있고 각각의 impact factor는 1, 2, 3, 4이다. P, Q 리스트는 각각 시작과 끝의 정보를 가진다고 생각하면 된다.
만약 CAGCCTA라는 DNS 시퀀스가 주어질 때 P[0] = 2, Q[0] = 4라면 DNS 시퀀스 중 2~4에 해당하는 값이 탐색 범위이다. (CAGCCTA)
탐색 범위 중 가장 작은 impact factor를 가지는 값을 찾아서 반환하면 된다.

문제 풀이

prefix sum으로 풀 수 있지만, 예외 처리로도 해당 문제를 풀 수 있다.

코드

def solution(S, P, Q):
    length = len(P)
    answer = []

    for i in range(length):
        prefix = S[P[i]:Q[i] + 1]
        try:
            prefix.index('A')
            answer.append(1)
        except ValueError:
            try:
                prefix.index('C')
                answer.append(2)
            except ValueError:
                try:
                    prefix.index('G')
                    answer.append(3)
                except ValueError:
                    answer.append(4)
    return answer