백준 15596번 정수 N개의 합
문제
정수 n개가 주어졌을 때 합을 구하는 함수를 작성한다. 언어마다 함수 형태가 정해져 있고, Python은 def solve(a: list) -> int다. a는 합을 구할 정수가 든 리스트(원소 0 이상 1,000,000 이하, 개수 1 이상 3,000,000 이하)이고, 반환값은 그 합이다. 입력을 받는 코드가 아니라 함수만 작성해서 제출한다.
풀이
ans를 0으로 두고 리스트를 for문으로 돌며 더한 뒤 반환한다.
함수를 오랜만에 써서 막막했고, Jupyter Lab에서 돌려 봐도 결과가 나오지 않았다. 함수는 정의만 해서는 실행되지 않고 직접 호출해야 값이 나온다. 이 문제는 채점 프로그램이 solve를 호출하므로 함수만 작성하면 되고, 제출 화면에도 아래와 같은 틀이 기본으로 들어 있었다.
def solve(a):
ans = 0
return ans빈 부분을 for문으로 채워 제출했다.
def solve(a):
ans = 0
for i in a:
ans += i
return ans짚고 갈 점
- 내장 함수
sum(a)로 한 줄에 풀 수도 있다.