用Python，用奇偶性判断幸运数字

Question

问题遇到的现象和发生背景

如果它的奇数位置的数字的和等于偶数位置的数字的和，那么我们称票为幸运的。
写下一个函数，它以一个票号作为参数，并返回最近的幸运票号（如果有两个的话）。
最接近的票号意味着它与给定的票号之间的绝对差异被最小化。
一个给定的票务号码应该是一个可接受的自然2𝑘位数（它有偶数位数），没有前导零。
笔记将程序保存到文件ticket.py中，并为解决方案函数提供名称get_nearest_lucky_ticket。

问题相关代码，请勿粘贴截图

Example
assert get_nearest_lucky_ticket ( 111111 ) == 111111
assert get_nearest_lucky_ticket ( 123321 ) == 123321
assert get_nearest_lucky_ticket ( 123320 ) == 123321
assert get_nearest_lucky_ticket ( 333999 ) == 334004

我想要达到的结果

代码最后使用以上assert测试代码，代码写一下注释。

Answer 1

def get_nearest_lucky_ticket(n):
    i = 0
    while i<=n:
        li = list(map(int,str(n+i)))
        a = sum(li[0::2])
        b = sum(li[1::2])
        if a==b:
            return n+i
        li = list(map(int,str(n-i)))
        a = sum(li[0::2])
        b = sum(li[1::2])
        if a==b:
            return n-i
        i += 1
assert get_nearest_lucky_ticket ( 111111 ) == 111111
assert get_nearest_lucky_ticket ( 123321 ) == 123321
assert get_nearest_lucky_ticket ( 123320 ) == 123321
assert get_nearest_lucky_ticket ( 333999 ) == 334004