백준 3, 5단계 코딩테스트

1. 더하기 사이클

문제: 0보다 크거나 같고, 99보다 작거나 같은 정수가 주어질 때 다음과 같은 연산을 할 수 있다. 먼저 주어진 수가 10보다 작다면 앞에 0을 붙여 두 자리 수로 만들고, 각 자리의 숫자를 더한다. 그 다음, 주어진 수의 가장 오른쪽 자리 수와 앞에서 구한 합의 가장 오른쪽 자리 수를 이어 붙이면 새로운 수를 만들 수 있다. 다음 예를 보자.

26부터 시작한다. 2+6 = 8이다. 새로운 수는 68이다. 6+8 = 14이다. 새로운 수는 84이다. 8+4 = 12이다. 새로운 수는 42이다. 4+2 = 6이다. 새로운 수는 26이다.

위의 예는 4번만에 원래 수로 돌아올 수 있다. 따라서 26의 사이클의 길이는 4이다.

N이 주어졌을 때, N의 사이클의 길이를 구하는 프로그램을 작성하시오.

입력: 첫째 줄에 N이 주어진다. N은 0보다 크거나 같고, 99보다 작거나 같은 정수이다.

출력: 첫째 줄에 N의 사이클 길이를 출력한다.

1. 내 풀이

처음에 시간초과가 뜨길래 sys.stdin.readline을 써줘야하나 싶었는데 마지막 if에서 숫자 a랑 문자열 tmp을 비교해서 무한루프에 빠진 것이었다...

import sys
a = int(sys.stdin.readline())
tmp = str(a)
cnt = 0

while True :
    if len(tmp) == 1 :
        tmp = tmp + tmp
        cnt += 1
    else :
        tmp2 = str(int(tmp[0]) + int(tmp[1]))
        if len(tmp2) == 2 :
            tmp = tmp[1] + tmp2[1]
        else :
            tmp = tmp[1] + tmp2[0]
        cnt += 1
    if a == int(tmp) :
        break
print(cnt)

2. 다른 사람의 풀이

파이썬에서만 쓸 수 있는 몫 구하기와 연산자랑 나머지 연산자를 사용해서 간결하게 풀 수도 있었다.

N = int(input())
num = 0
M = N
while True:
    A = N // 10
    B = N % 10
    N = 10*B + (A+B)%10
    num += 1
    if N == M:
        break
print(num)

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

1. 내 풀이

함수 구현만 하면 되는 문제였다.

def solve(a):
    ans = 0
    ans = sum(a)
    return ans

3. 셀프 넘버

문제: 셀프 넘버는 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보다 작거나 같은 셀프 넘버를 한 줄에 하나씩 증가하는 순서로 출력한다.

1. 내 풀이

예제에서는 셀프 넘버가 9993까지 나오는 걸로 되어있는데 나는 9995, 9997, 9999 3개가 더 나오길래 보니까 while문에서 a를 cnt으로 바꾸니까 제대로 출력되었다.
그나저나 코드가 많이 난잡하다...

li = []
for i in range(1, 10001) :
    li.append(i)

def solve(a):
    ans = 0
    if a < 10 :
        ans = a+a
    else :
        a = str(a)
        ans = int(a)
        for i in range(len(a)) :
            ans += int(a[i])
    return ans

cnt = 1

while True :
    a = solve(cnt)
    if cnt > 10000 :
        break
    else :
        cnt += 1
        if a in li :
            li.remove(a)

for i in range(len(li)) :
    print(li[i])

2. 다른 사람의 풀이

• set함수를 사용해서 기본 배열에서 생성자가 있는 숫자배열을 빼서 리턴해주는 방식을 사용했다. (생성자 숫자배열은 중복이 있을 수 있으니 set함수를 사용한 것 같다)

• range함수를 쓰면 한 줄로 순차배열을 만들 수 있음

  range(start, stop, step(옵션))

def d(n):
    n=str(n)
    sum=int(n)
    for i in range(len(n)):
        sum+=int(n[i])
    return sum
    
a=set(range(1,10001))
b=set()

for i in range(1,10001):
    b.add(d(i))
    
c=a-b
c=list(c)
c.sort()

for i in range(len(c)):
    print(c[i])