[HackerRank] Repeated String

[문제 링크]

[입력]

s: 반복할 문자열
n: 고려해야 될 알파벳의 개수

[출력]

int: 문자열중 'a' 의 개수

[코드]

def repeatedString(s, n):
    # Write your code here
    set_of_string = list(set(s))
    if(len(set_of_string)==1 and set_of_string[0]=='a'):
        return n
    result = 0
    for alpha in s:
        if(alpha == 'a'):
            result +=1
    result = result * (n//len(s))
    for alpha in s[:(n%len(s))]:
        if(alpha == 'a'):
            result +=1
    return result