백준 단계별로 문제풀기
6단계 : 함수
문제 목록
1. 15596번 : 정수 n개의 합
2. 4673번 : 셀프넘버
1. 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개의 합 (정수)
🙋♀️ 내코드
def solve(a):
return sum(a)
2. 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보다 작거나 같은 셀프 넘버를 한 줄에 하나씩 증가하는 순서로 출력한다.
🙋♀️ 내 코드
cnt = round((10/2)+(9980/11))
i = -1
i_list = []
for _ in range(cnt):
if i < 8 :
i += 2
else :
i+= 11
i_list.append(i)
print(i_list[:20],sep='\n')
원래는 셀프넘버의 규칙이 10 이전에는 2씩 증가하고,
이후에는 11씩 증가하는 줄 알고 그렇게 코드를 짰다.
그러나 아닌가보다.
리스트를 만들고 뒤의 [:20]을 출력하니 달랐다..
분명 이걸 반영하는 수열이 있을 것 같아서 찾아볼까 고민하다 문득 깨달았다.
이 문제의 의도는 뭘까?
내가 이문제를 통해 알아야 하는 것은 무엇인가?
왜냐하면 이 '함수'문제세트에서 첫번째 문제가 너무 쉬웠다. def 이용법을 익히는 것이었다.
그렇다면 이번 문제는?
클래스의 self를 이용하는 법을 아는 것이 목표 아닐까?
왜냐면 처음에 생성자 법칙도 재귀함수처럼, 결과값을 다시 가져와서 해야한다.
그러나 여기서 끝나지 않고, 이 결과들이 해당되지않는 차집합을 구하는 것이 이 문제의 정답이다.
그러므로,
- 생성자가 있는 수열을 구한다.
- 10000까지의 집합 중 1.에서 구한 수열을 뺀 차집합을 구한다.
이 두 과정을 거치는 코드를 만들어야 한다.
이 과정에서 '이전에 추정한 문제 의도'에 맞게 클래스와 self, init을 써보겠다 (써보려고 했는데 쓸모 없어서 공부만 하고 안씀, 공부했으니 이득이다!)
🙋♀️ 다시 도전
s = set()
nn = set(range(1,10001))
def d(n):
num = n + sum(map(int,str(n)))
s.add(num)
for i in range(1,10001):
if d(i) not in nn :
print(i)더 짧게 / 빠르게 하고 싶어 도전! (했으나 속도도 쓰는 메모리도 같다..)
def d(n):
num = n + sum(map(int,str(n)))
return num
nn = set(range(1,10001))
for i in range(1,10001):
nn.discard(d(i))
print(*nn,sep ='\n')