Ускорить код.Задача кубики

an, bor = list(map(int, input().split()))
a = set()
b = set()
for i in range(an):
    a.add(int(input()))
for j in range(bor):
    b.add(int(input()))
print(len(a.intersection(b)))
print(*sorted(a.intersection(b)))
print(len(a.difference(b)))
print(*sorted(a.difference(b)))
print(len(b.difference(a)))
print(*sorted(b.difference(a)))

N,M =[int(i) for i in input().split()]
kub_anna = set(int(input()) for _ in range(N))
kub_boris = set(int(input()) for _ in range(M))
for i in (kub_anna & kub_boris,kub_anna - kub_boris,kub_boris - kub_anna):
    print(len(i))
    print(*(sorted(i)))

    s1 = a.intersection(b)
    print(len(s1))
    print(' '.join(map(str, sorted(s1))))
    s2 = a.difference(b)
    print(len(s2))
    print(' '.join(map(str, sorted(s2))))
    s3 = b.difference(a)
    print(len(s3))
    print(' '.join(map(str, sorted(s3))))

def print_set(some_set):
    print(len(some_set))
    print(*[str(item) for item in sorted(some_set)])
 
N, M = [int(s) for s in input().split()]
A_colors, B_colors = set(), set()
for i in range(N):
    A_colors.add(int(input()))
for i in range(M):
    B_colors.add(int(input()))
    
print_set(A_colors & B_colors)
print_set(A_colors - B_colors)
print_set(B_colors - A_colors)

s1 = a.intersection(b)
s2 = a.difference(s1)
s3 = b.difference(s1)

def f6(aa,bb):
    a = set(aa)
    b = set(bb)
    def print1(s):
        print(len(s))
        w = list(map(str, sorted(map(int, s))))
        w2 = ' '.join(w)
        print(w2)
    s1 = a.intersection(b)
    s2 = a.difference(s1)
    s3 = b.difference(s1)
    print1(s1)
    print1(s2)
    print1(s3)