[Python] 1부터 N까지 합 구하기

첫 번째 방법

def sum_n(n):
    sum = 0
    for i in range(n+1):
        sum += i
    return sum

print(sum_n(10))
print(sum_n(100))

흔히 알고 있는 sum을 이용해 1부터 N까지의 합을 구하는 프로그램이다.

사실 다른 방법은 생각도 못해봤는데 가우스의 합 공식을 이용하면 더 간단하게 풀 수 있다.

 

두 번째 방법

def sum_n(n):
    return n * (n + 1) //2

print(sum_n(10))
print(sum_n(100))

코드가 훨씬 간단해졌다. 

역시 수학을 알면 알수록 알고리즘에 많은 도움이 되는 것 같다.