백준 2355번: 시그마

문제

문제 바로가기> 백준 2355번: 시그마

풀이

1부터 n까지 합의 공식을 이용하면 쉽게 풀 수 있다. 이때 a가 b 보다 큰 경우도 고려해주어야한다.

def solution():
    import sys
    input = sys.stdin.readline
    a, b = map(int, input().split())
    if a<b:
        asum, bsum = a*(a-1)//2, b*(b+1)//2
        print(bsum-asum)
    else:
        asum, bsum = a*(a+1)//2, b*(b-1)//2
        print(asum-bsum)
solution()