Как оптимизировать программу?

n = int(input())
n1 = ""
n2 = ""
res = 0
res1 = 0
res2 = 0
if n%2 != 0 and n != 1:
    for i in range(10**(n-1), 10**n):
        n1 = str(i)[:n//2]
        n2 = str(i)[n//2+1:]
        for j in range(len(n1)):
            res1 += int(n1[j])
            res2 += int(n2[j])
        if res1 == res2:
            res += 1
            print(i, n1, n2, res1, res2, res)
        res1 = 0 
        res2 = 0
elif n%2 == 0 and n != 1:
    for i in range(10**(n-1), 10**n):
        n1 = str(i)[:n//2]
        n2 = str(i)[n//2:]
        for j in range(len(n1)):
            res1 += int(n1[j])
            res2 += int(n2[j])
        if res1 == res2:
            res += 1
            print(i, n1, n2, res1, res2, res)
        res1 = 0 
        res2 = 0
else:
    res = 9
print(res)

n = int(input())
p = n // 2 + 1 if n % 2 else n // 2
 
s = (str(i) for i in range(10 ** (n - 1), 10 ** n))
c = 0
for v in s:
    if sum(map(int, v[:n // 2])) == sum(map(int, v[p:])):
        c += 1
        m = sum(map(int, v[:n // 2]))
        print(v, v[:n // 2], v[p:], m, m, c)
print(c)

n = int(input())
p = 10 ** (n // 2 + 1) if n % 2 else 10 ** (n // 2)
 
c = 0
for i in range(10 ** (n - 1), 10 ** n):
    a = a_t = i // p
    b = b_t = i % 10 ** (n // 2)
    
    a_s = 0
    while a_t:
        a_s += a_t % 10
        a_t //= 10
    
    b_s = 0
    while b_t:
        b_s += b_t % 10
        b_t //= 10
        
    if a_s == b_s:
        c += 1
        print(i, a, b, a_s, b_s, c)
        
print(c)

def is_lucky(ticket_nums):
    len_str = len(ticket_nums)
    if len_str == 2:
        return ticket_nums[0] == ticket_nums[1]
    half = len_str // 2
    return sum(map(int, ticket_nums[:half])) == sum(map(int, ticket_nums[half:]))