[백준 15596 / 파이썬(Python) / Bronze Ⅱ] 구현 / 정수 N개의 합

문제링크 : https://www.acmicpc.net/problem/15596

15596번: 정수 N개의 합

 

■ 문제

정수 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개의 합 (정수)

■ 풀이

• 리스트 합을 구하는거 같아서 파이썬 내장함수 sum()을 사용함

# 코드 1
def solve(a):
    
    return sum(a)

# 코드 2
def solve(a):
    ans = 0
    for i in a:
        ans += i
    return ans

확인용 코드

def solve(a):
    ans = 0
    for i in a:
        ans += i
    return ans
n = list(map(int, input().split()))
solve(n)

# 30