[항해]알고리즘 스터디(백준 #1110)

백준#1110

바로가기

문제
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의 사이클의 길이를 구하는 프로그램을 작성하시오.

입출력 규칙
1. 입력
첫째 줄에 N이 주어진다. N은 0보다 크거나 같고, 99보다 작거나 같은 정수이다.
1. 출력
첫째 줄에 N의 사이클 길이를 출력한다.

코드
1안

#1. 정답코드 (int 계산)
import sys
n = int(sys.stdin.readline()) # // int 68
num = n
cnt = 0
while True:
    a = num // 10 # // int 6
    b = num % 10 # // int 8
    c = (a+b) % 10 # // int 1"4" == 나머지 4
    num = (b*10) + c # // 80 + 4
    cnt += 1
    if(num == n):
        break
print(cnt)

중요)
1) sys.stdin.readline : 빠른 입력!
input()과 겉보기에 똑같은 역할을 하지만, 시간초과가 나는 경우 빠른입력 사용하면 효율 상승한다. 다만 input()과는 다르게 '사용자가 입력한 문자열 + 개행 문자(\n)'가 저장되기 때문에 str문에서는 str(sys.stdin.readline.split('\n')) 이용하면 좋다.
2) while (조건) : 조건이 참(True)인 동안 이하문을 반복해서 실행한다! 정답코드를 보면 while True: 가 있음. 또한 if문도 포함되어 있음. 즉, 계속 while 이하 열을 반복하다가 if문의 조건 맞아서 실행(break)되면 멈추는 구조임.
2안

#2. 시간초과 코드 (str 계산)
import sys
n = str(sys.stdin.readline().strip('\n'))
num = n
cnt = 0
while True:
  if len(num) == 1:
      num = "0" + num
  plus = str(int(num[0]) + int(num[1]))  # str '8'
  num = num[-1] + plus[-1]  # str '68' #[-1]
  cnt += 1
  if num == n:
      print(cnt)
      break

중요)
1) str 계산한 결과 시간초과... : 정답코드와 다르게 아래는 시간초과로 판단되었다. 정답코드는 int로 계산을 하기 때문에 시간초과 코드에 비해 계산이 모든 면에서 빨랐다.
출력하는 결과는 같더라도 시간 차이가 크게 났다.
2) plus 코드에서 [-1]은 뭘까? : 제일 오른쪽 1자리만을 선택하는 표현임. 위 코드 내에서 계산되어야 하는 코드는 모두 '두자리'인데, [-1] 처리를 안할 경우 plus 계산값이 그대로 num 계산식에 반영되어 '두자리'로 반환되어야 할 num값이 '세자리'가 되는 경우가 발생한다. (ex : plus가 '14'일 때 num=num[-1]+plus[-1]일때는 num[-1]의 끝자리 '8'과 plus의 끝자리인 '4'가 식에 반영되어 '84'를 반환해야 하나, plus[-1]이 아니라 plus일 경우 '14'모두가 반영되어 '814'를 반환하므로 결과적으로 오류!