전공/백준
14650
yha97
2022. 11. 6. 23:45
날짜 : 2022. 11. 06
사용 언어 : python
문제
코드
import sys
n = int(sys.stdin.readline())
num = [0, 1, 2]
total = []
total.append([]) # 임시
total.append([0, 1, 2]) # 1자리
res = 0
def check(n): # 3의 배수 확인
t = 0
while n > 0:
t += n % 10
n = n // 10
if t % 3 == 0 and t > 0: return 1
else: return 0
for i in range(2, n+1): # 2자리부터 체크
temp = []
for j in range(len(total[i-1])):
for k in range(3):
t = total[i-1][j] * 10 + num[k] # 0, 1, 2에서 확인
if t >= (10 ** (i-1)): # 해당 자릿수의 최솟값과 비교해서 push
temp.append(t)
total.append(temp)
#print(total) # 디버그
for i in total[n]:
res += check(i) # 3의 배수인지 해당 자릿수에서 체크
print(res)
알게된 점
- 먼저 2차원 배열을 생성하고 해당 row 자리마다 해당되는 자릿수에 맞는 숫자를 만들었다.
- 그리고 그 수들이 3의 배수인지 체크하고 그 개수를 세어 아웃풋을 출력했다.
참고 사이트
-