자연수 n을 뒤집어 각 자리 숫자를 원소로 가지는 배열 형태로 리턴해주세요. 예를들어 n이 12345이면 [5,4,3,2,1]을 리턴합니다.
def solution(n):
answer = []
while n:
answer.append(n % 10)
n //= 10
return answer
def solution(n):
n = str(n)
answer = []
for i in n:
answer.append(int(i))
return answer[::-1]
# Test the function directly
print(solution(12345))
[::-1]
을 사용하면 리스트가 뒤집힌다는 것을 알았습니다.def solution(n):
return [int(i) for i in str(n)][::-1]
print(solution(12345)) # 출력: [5, 4, 3, 2, 1]
[int(i) for i in str(n)]
이라는 표현식 하나로 문자열 변환, 순회, 정수 변환을 동시에 수행할 수 있습니다.[::-1]
을 리스트 컴프리헨션과 함께 사용하면, 리스트를 뒤집을 수 있다는 것을 배웠습니다.def solution(n):
return list(map(int, reversed(str(n))))
print(solution(12345)) # 출력: [5, 4, 3, 2, 1]
reversed()
는 문자열을 뒤집고, map()
은 각 문자를 정수로 변환합니다.