우당탕탕 파이썬 with 하루하나 알고리즘

백준 15596 파이썬 - 함수 생성하기

hyxndy 2022. 4. 1. 09:26

백준 15596번 - 정수 N개의 합 with Python

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

https://www.acmicpc.net/problem/15596

 

15596번: 정수 N개의 합

C++17, Java 8, Python 3, C11, PyPy3, C99, C++98, C++11, C++14, Go, C99 (Clang), C++98 (Clang), C++11 (Clang), C++14 (Clang), C11 (Clang), C++17 (Clang)

www.acmicpc.net


파이썬 함수 구조는 다음과 같다.

def 함수명(매개변수):
    <수행할 문장1>
    <수행할 문장2>
    ...

이 구조를 이용해 리스트 a에 포함되어 있는 정수 n개의 합을 구하면 된다.

먼저 함수 이름을 지정해주어야 하는데, solve(a)로 지정하겠다.

 

리스트 내에 포함되어 있는 정수의 합을 구하는 파이썬 내장함수는 sum(리스트)이다.

그래서 수행할 문장의 위치에 sum(a)를 입력해주면 된다.

 

이런식으로 코드를 작성하게 되면 solve(a) 를 실행시키면

자동으로 a리스트의 정수의 합을 구할 수 있게 될 것이다.

 

<코드>

def solve(a):
	sum(a)