Problem link

My solution that is supposed to get a TLE but got AC instead

Generator script for the test case that could trigger a TLE:

from math import isqrt
N=10**7
sieve=[True]*(N+1)
for i in range(2,isqrt(N)+1):
    if not sieve[i]:continue
    for j in range(i**2,N+1,i):
        sieve[j]=False
sieve[0]=sieve[1]=False
primes=[x for x in range(2,N) if sieve[x]]
print(1)
print(len(primes),1)
print(*primes)