[PS] BOJ 1065 / 한수
문제 링크: https://www.acmicpc.net/problem/1065
Thumbnail: Photo by Lanju Fotografie / Unsplash
숫자의 범위가 작아서, 비교적 간단한 방법으로도 풀리는 문제였습니다.
문제
어떤 양의 정수 \(X\)의 각 자리가 등차수열을 이룬다면, 그 수를 한수 라고 한다.
- 등차수열은 연속된 두 개의 수의 차이가 일정한 수열을 말한다.
\(N\)이 주어졌을 때, \(1 \le X \le N\) 인 한수의 개수를 출력하는 프로그램을 작성하시오.
입력
첫째 줄에 1,000보다 작거나 같은 자연수 N이 주어진다.
출력
첫째 줄에 1보다 크거나 같고, N보다 작거나 같은 한수의 개수를 출력한다.
풀이
1보다 크고 \(N\)보다 작은 임의의 \(X\)가 한수인지 아닌지 판단하는 과정을 생각해보자.
먼저, \(X\)가 두 자리 이하의 수인 경우에는 항상 한수라고 판단할 수 있다.
두 자리 이하의 수인 경우는 X의 각 자리가 언제나 등차수열이기 때문이다!
이제 세 자리 이상의 수 X가 한수인지 판단하는 과정을 코드로 구현하면 된다.
나
코드
from sys import stdin
N: int = int(stdin.readline().strip())
cnt: int = 0
def isHansu(x: int) -> bool:
"""입력으로 받은 수가 한수인지 아닌지 판별한다.
Args:
x (int): 1 이상 N(<=1000) 이하의 정수
Returns:
bool: x가 한수이면 True, 아니면 False
"""
if x < 100: # 두 자리 이하의 수는 언제나 각 자리가 등차수열을 이룬다!
return True
else:
x_str = str(x)
diffs = []
for i in range(1, len(x_str)):
diffs.append(int(x_str[i]) - int(x_str[i - 1]))
for i in range(1, len(diffs)):
if diffs[i] != diffs[i - 1]:
return False
return True
for i in range(1, N + 1):
if isHansu(i):
cnt += 1
print(cnt)
solution.py
👀 이해의 편의를 위해 타입 힌트를 적용한 코드이나, 실제로 제출 시에는 타입 힌트를 지우고 제출하는 것을 권장드립니다.
타입 힌트는 파이썬 코드의 실행 속도를 늦추는 원인 중 하나입니다.