[Baekjoon/Python3] 1292번 쉽게 푸는 문제

[Baekjoon/Python3] 1292번 쉽게 푸는 문제

 

www.acmicpc.net/problem/1292

1292번: 쉽게 푸는 문제

 

def func(n):
    res = 0
    cnt = 0
    for i in range(1, n + 1):
        for j in range(i):
            if cnt < n:
               res += i
               cnt += 1
            else:
                break
    return res
a, b = map(int, input().split())
print(func(b) - func(a - 1))