Problem p02574 (python)

Fixed submission:
import math
(_,), a  = [list(map(int, s.split())) for s in open(0)]

def fast_factorize(num):
    min_factor_list = list(range(num + 1))
    for i in range(int(math.sqrt(num) + 1), 1, -1):
        for j in range(i, num+1, i):
            min_factor_list[j] = i
    return min_factor_list

min_factor_list = fast_factorize(max(a))

prime_factor_sets = []
for x in a:
    prime_factor_set = set()
    while min_factor_list[x] != 1:
        prime_factor_set.add(min_factor_list[x])
        x //= min_factor_list[x]
    prime_factor_sets.append(prime_factor_set)

g = prime_factor_sets[0].copy()
l = prime_factor_sets[0].copy()
s = 0
for x in prime_factor_sets:
    g &= x
    l |= x
    s += len(x)

if len(g) != 0:
    ans = 'not coprime'
elif len(l) != s:
    ans = 'setwise coprime'
else:
    ans = 'pairwise coprime'
print(ans)

Buggy submission (verdict: Runtime Error):
import math
(_,), a  = [list(map(int, s.split())) for s in open(0)]

def fast_prime_factorize(num):
    prime_factor_sets = [set() for _ in range(num+1)]
    for i in range(2, int(math.sqrt(num)) + 2):
        if len(prime_factor_sets[i]) == 0:
            for j in range(i, num+1, i):
                prime_factor_sets[j].add(i)
    return prime_factor_sets

def get_prime_factor_set(prime_factor_sets, x):
    y = prime_factor_sets[x]
    if len(y) == 0 and x != 1:
        y = {x}
    return y


prime_factor_sets = fast_prime_factorize(max(a))

g = get_prime_factor_set(prime_factor_sets, a[0]).copy()
l = set()
s = 0
for x in a:
    y = get_prime_factor_set(prime_factor_sets, x)
    g &= y
    l |= y
    s += len(y)

if len(g) != 0:
    ans = 'not coprime'
elif len(l) != s:
    ans = 'setwise coprime'
else:
    ans = 'pairwise coprime'
print(ans)

Buggy submission (verdict: Runtime Error):
import math
(_,), a  = [list(map(int, s.split())) for s in open(0)]

def fast_prime_factorize(num):
    prime_factor_sets = [set() for _ in range(num+1)]
    for i in range(2, int(math.sqrt(num)) + 2):
        if len(prime_factor_sets[i]) == 0:
            for j in range(i, num+1, i):
                prime_factor_sets[j].add(i)
    return prime_factor_sets

prime_factor_sets = fast_prime_factorize(max(a))

x = a[0]
y = prime_factor_sets[x]
if len(y) == 0 and x != 1:
    y = {x}
g = y.copy()
l = set()
s = 0
for x in a:
    y = prime_factor_sets[x]
    if len(y) == 0 and x != 1:
        y = {x}
    g &= y
    l |= y
    s += len(y)

if len(g) != 0:
    ans = 'not coprime'
elif len(l) != s:
    ans = 'setwise coprime'
else:
    ans = 'pairwise coprime'
print(ans)

Buggy submission (verdict: Runtime Error):
import math
(_,), a  = [list(map(int, s.split())) for s in open(0)]

def fast_prime_factorize(num):
    prime_factor_sets = [set() for _ in range(num+1)]
    for i in range(2, int(math.sqrt(num)) + 2):
        if len(prime_factor_sets[i]) == 0:
            for j in range(i, num+1, i):
                prime_factor_sets[j].add(i)
    return prime_factor_sets

prime_factor_sets = fast_prime_factorize(max(a))

y = prime_factor_sets[a[0]]
if len(y) == 0 and a[0] != 1:
    y = {a[0]}
g = y.copy()
l = set()
s = 0
for x in a:
    y = prime_factor_sets[x]
    if len(y) == 0 and x != 1:
        y = {x}
    g &= y
    l |= y
    s += len(y)

if len(g) != 0:
    ans = 'not coprime'
elif len(l) != s:
    ans = 'setwise coprime'
else:
    ans = 'pairwise coprime'
print(ans)

Buggy submission (verdict: Runtime Error):
import math
(_,), a  = [list(map(int, s.split())) for s in open(0)]

def fast_prime_factorize(num):
    prime_factor_sets = [set() for _ in range(num+1)]
    for i in range(2, int(math.sqrt(num)) + 2):
        if len(prime_factor_sets[i]) == 0:
            for j in range(i, num+1, i):
                prime_factor_sets[j].add(i)
    return prime_factor_sets

prime_factor_sets = fast_prime_factorize(max(a))

g = prime_factor_sets[0].copy()
l = set()
s = 0
for x in a:
    y = prime_factor_sets[x]
    if len(y) == 0 and x != 1:
        y = {x}
    g &= y
    l |= y
    s += len(y)

if len(g) != 0:
    ans = 'not coprime'
elif len(l) != s:
    ans = 'setwise coprime'
else:
    ans = 'pairwise coprime'
print(ans)