Language/Python

[BAEKJOON]백준 15596번: 정수 N개의 합

림밤빵 2021. 2. 8. 15:33
728x90

문제

정수 n개가 주어졌을 때, n개의 합을 구하는 함수를 작성하시오.

작성해야 하는 함수는 다음과 같다.

  • Python 2, Python 3, PyPy, PyPy3: def solve(a: list) -> int
    • a: 합을 구해야 하는 정수 n개가 저장되어 있는 리스트 (0 ≤ a[i] ≤ 1,000,000, 1 ≤ n ≤ 3,000,000)
    • 리턴값: a에 포함되어 있는 정수 n개의 합 (정수)

 

코드

def solve(a):
    return sum(a)

if __name__ == '__main__':
    numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
    print(solve(numbers))

코드 실행 결과

 

코드 설명

매개변수로 a를 받는 solve함수를 정의하고 a의 합을 반환 해주도록 하였다.

main함수에서 1부터 10까지 10개의 정수를 가지고 있는 리스트 numbers를 정의하였다.

solve(numbers)를 출력하여 numbers의 합을 출력하도록 하였다.

def solve(a):
    return sum(a)

if __name__ == '__main__':
    numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
    print(solve(numbers))
728x90