Python으로 배열의 합 쉽게 구하기

이 글에서는 배열의 합을 구하는 문제와 그 해결 방법을 단계별로 살펴보겠습니다.

문제 정의

문제 − 하나의 배열이 주어졌을 때, 배열에 포함된 모든 요소의 합을 계산하는 프로그램을 작성합니다.

가장 기본적인 접근 방식(브루트 포스)은 배열 전체를 처음부터 끝까지 순회하면서 각 인덱스의 값을 하나씩 더해 합을 구하는 것입니다. Python에서는 이 과정을 반복문으로 직접 구현할 수도 있고, 내장 함수 sum()을 활용하면 한 줄로 간단하게 처리할 수도 있습니다.

방법 1: 내장 함수 sum() 사용

# sum 함수 정의
def sum_(arr, n):
    # 내장 함수 사용
    return(sum(arr))
# main
arr = [11, 22, 33, 44, 55, 66]
# 배열 길이
n = len(arr)
ans = sum_(arr, n)
# 결과 출력
print("배열의 합은", ans)

실행 결과

배열의 합은 231

위 코드에서 sum_(arr, n) 함수는 배열과 그 길이를 매개변수로 받아, 내장 함수 sum()을 호출해 모든 요소의 합을 반환합니다. 함수 내에서 선언된 변수들은 모두 지역 범위(local scope)에 속하며, 함수 호출이 끝나면 소멸합니다.

방법 2: 반복문으로 직접 구현

내장 함수를 사용하지 않고 반복문으로 직접 합을 구할 수도 있습니다.

def sum_loop(arr):
    total = 0
    for num in arr:
        total += num
    return total

arr = [11, 22, 33, 44, 55, 66]
print("배열의 합은", sum_loop(arr))

이 방식은 배열의 각 요소를 순서대로 방문하며 total 변수에 누적하는 원리로 동작합니다. 시간 복잡도는 O(n)으로, 배열의 크기에 비례합니다.

결론

이 글에서는 Python에서 배열의 합을 구하는 두 가지 방법, 즉 내장 함수 sum()을 활용하는 방법과 반복문으로 직접 누적하는 방법을 배웠습니다. 실무에서는 코드의 간결성과 가독성을 위해 내장 함수를 사용하는 것이 일반적으로 권장되며, 학습 목적이나 로직 제어가 필요한 경우에는 반복문 방식이 유용합니다.