luoheng
9/4/2019 - 7:25 AM

GenomicRangeQuery

# you can write to stdout for debugging purposes, e.g.
# print("this is a debug message")

def _cal(preSum, count):
    if preSum[0] > count[0]:
        return 1
    elif preSum[1] > count[1]:
        return 2
    elif preSum[2] > count[2]:
        return 3
    else:
        return 4


def solution(S, P, Q):
    # write your code in Python 3.6
    d = {'A':0, 'C':1, 'G':2, 'T':3}
    preSum = [[0, 0, 0, 0] for _ in range(len(S))]
    for i, s in enumerate(S):
        preSum[i] = preSum[i-1][:]
        preSum[i][d[s]] += 1
    result = [0] * len(P)
    for i in range(len(P)):
        x, y = P[i], Q[i]
        if x == 0:
            count = [0, 0, 0, 0]
        else:
            count = preSum[x-1]
        result[i] = _cal(preSum[y], count)
    return result