백준 단계별 - 함수

<백준 15596번 - 정수 N개의 합> https://www.acmicpc.net/problem/15596

○문제

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

import sys

def solve(a):
    ans = 0
    for i in a:
        ans += i
    return ans

a = list(map(int, sys.stdin.readline().split()))

- 오류: print(solve(a))

채점할 때 function을 자동으로 실행하기 때문에 따로 실행문까지 구현할 필요가 없다.

---

<백준 4673번 - 셀프 넘버> https://www.acmicpc.net/problem/4673

○문제

셀프 넘버는 1949년 인도 수학자 D.R. Kaprekar가 이름 붙였다. 양의 정수 n에 대해서 d(n)을 n과 n의 각 자리수를 더하는 함수라고 정의하자. 예를 들어, d(75) = 75+7+5 = 87이다.

양의 정수 n이 주어졌을 때, 이 수를 시작해서 n, d(n), d(d(n)), d(d(d(n))), ...과 같은 무한 수열을 만들 수 있다. 

예를 들어, 33으로 시작한다면 다음 수는 33 + 3 + 3 = 39이고, 그 다음 수는 39 + 3 + 9 = 51, 다음 수는 51 + 5 + 1 = 57이다. 이런식으로 다음과 같은 수열을 만들 수 있다.

33, 39, 51, 57, 69, 84, 96, 111, 114, 120, 123, 129, 141, ...

n을 d(n)의 생성자라고 한다. 위의 수열에서 33은 39의 생성자이고, 39는 51의 생성자, 51은 57의 생성자이다. 생성자가 한 개보다 많은 경우도 있다. 예를 들어, 101은 생성자가 2개(91과 100) 있다. 

생성자가 없는 숫자를 셀프 넘버라고 한다. 100보다 작은 셀프 넘버는 총 13개가 있다. 1, 3, 5, 7, 9, 20, 31, 42, 53, 64, 75, 86, 97

10000보다 작거나 같은 셀프 넘버를 한 줄에 하나씩 출력하는 프로그램을 작성하시오.

○입력

입력은 없다.

○출력

10,000보다 작거나 같은 셀프 넘버를 한 줄에 하나씩 증가하는 순서로 출력한다.

num_list=list(range(1,10001))

def d(n):
  d_num=list(str(n)) # 자릿수를 분리해서 리스트에 담음
  d_num=list(map(int,d_num)) # 리스트에 담긴 문자열을 정수로 형변환
  init_num = n #생성자가 있는 수

  for i in d_num:
    init_num+=i #n+각 자릿수의 숫자 합 -> 다음 수

  return init_num

for n in range(1,10001):   
  if d(n) in num_list:
    num_list.remove(d(n)) #생성자가 있는 수(init_num) 삭제

for i in num_list:
    print(i)

생성자의 개념을 이해하는 데 시간이 걸렸다. ini_num 자체를 생성자로 보아서 생성자가 없는 셀프 넘버를 출력하기 어려웠다. init_num은 각 자리수를 합해서 나오는 수이기 때문에 생성자가 무조건 있는! 수라는 점이 핵심이었다. 

---

<백준 1065번 - 한수> https://www.acmicpc.net/problem/1065

○문제 

어떤 양의 정수 X의 각 자리가 등차수열을 이룬다면, 그 수를 한수라고 한다. 등차수열은 연속된 두 개의 수의 차이가 일정한 수열을 말한다. N이 주어졌을 때, 1보다 크거나 같고, N보다 작거나 같은 한수의 개수를 출력하는 프로그램을 작성하시오. 

○입력

첫째 줄에 1,000보다 작거나 같은 자연수 N이 주어진다.

○출력

첫째 줄에 1보다 크거나 같고, N보다 작거나 같은 한수의 개수를 출력한다.

n = int(input())
num_list = list(range(100,n+1))
result_num = []
for i in num_list:
    arith_num=list(str(i)) 
    arith_num=list(map(int,arith_num)) 
    if arith_num[0]- arith_num[1] == arith_num[1]- arith_num[2] :
        result_num.append(i)

if n>=100:
    print(99+len(result_num))
else:
    print(n)

- 오류: print(99+len(result_num))

n이 한 자리 수일 경우에도 99개를 더했다. 두 자리수일 때와 두 자리 수일 때를 분리할 필요가 있다.

~ 수정:
if n>=100:
    print(99+len(result_num))
else:
    print(n)

---

코딩테스트를 풀면서 느끼는 건데, list를 잘 사용하면 반은 가는 것 같다.

list로 입력을 받는 방법이 익숙해지도록 연습해야겠다.