백준 1292 (python): 쉽게 푸는 문제

A, B = map(int, input().split())

def solution(number):
    num = 1
    answer = 0
    i = 0
    for _ in range(number):
        answer += num
        i += 1
        if num == i:
            num += 1
            i = 0
    return answer

print(solution(B) - solution(A-1))

 

처음에 겁 먹었는데 생각보다 간단했다.

그냥 하라는대로 하기