[프로그래머스] / [Level 2] / [Python] 우박수열 정적분

https://school.programmers.co.kr/learn/courses/30/lessons/134239

프로그래머스

---

수열을 계산하면서 사다리꼴 넓이를 계산한다.

def collatz(n):
    area = []    
    idx = 0
    
    while n > 1:
        pn = n
        
        if n % 2 == 0:
            n = n // 2
        else:
            n = n * 3 + 1
        
        idx += 1

        area.append((n + pn)/2.0)
    
    return area
        

def solution(k, ranges):
    answer = []
    
    area = collatz(k)
    
    for range in ranges:
        a = range[0]
        b = (len(area) + 1) + range[1]
        
        if a < b:
            answer.append(sum(area[a:b-1]))
        else:
            answer.append(-1.0)

    return answer