a가 2부터 x까지 돌기 때문에 시간초과가 뜬 거 같습니다.

a*a가 x보다 작거나 같을 때까지만 반복문을 돌리시면 시간초과가 안 뜰 거 같습니다.

함수가 문제인줄 알고 아래의 코드를 짜서 시도했는데 시간이 들긴 해도 해결됬습니다

알려주셔서 감사합니다

x = int(input())

a = 2

while True:

if x % a == 0:
print(a)
x = x//a
continue
if x % a != 0:
a += 1
if x == 1:
break