Alien Invasion II

""" https://cses.fi/345/task/A"""

def solve(n):
    sum_digits = sum([int(d) for d in n])
    need = 3 - (sum_digits % 3)  # How many to get a 3-divisible number
    print(str(need) + n)
    not_prime = int(str(need) + n)
    print(f"{not_prime // 3} {3}")


num = input()
solve(num)

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED