2018년 2월 6일 화요일

[python][hackerrank] Viral Advertising

좀 더 단순하게 풀 수는 없었을까?
날짜가 지나갈 때마다 생존하는 녀석들을 따로 기억해야했고
그거와 별개로 내가 뱉어내야할 결과값을 따로 담아야했다.
def viral_ad(n):
    tmp = 2
    res = 2
    while n > 1:
        tmp = (tmp * 3) // 2
        res += tmp
        n -= 1
    return res

n = int(input().strip())
result = viral_ad(n)
print(result)

댓글 없음:

댓글 쓰기