Alkuluvut

def on_alkuluku(luku):
    if luku == 1:
        return False
    if luku == 2:
        return True
    for i in range(2, luku // 2):
        if luku % i == 0:
            return False
    return True

def tee_alkuluku(luku, numerot):
    if len(luku) > 17:
        return 0
    if len(luku) > 0 and on_alkuluku(int(luku)):
        return luku
    for i in range(0, len(numerot)):
        luku = luku + numerot[i];
        ret = tee_alkuluku(luku, numerot)
        if ret != 0:
            return ret;
        luku = luku[:-1]
    return 0

def main():
    t = int(input())
    tuloste = ""
    for i in range(t):
        k = int(input())
        rivi = input()
        numerot = rivi.split()
        alkuluku = int(tee_alkuluku("", numerot))
        if alkuluku == 0:
            tuloste = tuloste + "NO\n"
        else:
            tuloste = tuloste + "YES\n"
            tuloste = tuloste + str(alkuluku) + "\n";
    print(tuloste)

main()

Test 1

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 2

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED