PAT1019 数字黑洞 python

题目
给定任一个各位数字不完全相同的 4 位正整数，如果我们先把 4 个数字按非递增排序，再按非递减排序，然后用第 1 个数字减第 2 个数字，将得到一个新的数字。一直重复这样做，我们很快会停在有“数字黑洞”之称的 6174，这个神奇的数字也叫 Kaprekar 常数。

例如，我们从6767开始，将得到

7766 - 6677 = 1089
9810 - 0189 = 9621
9621 - 1269 = 8352
8532 - 2358 = 6174
7641 - 1467 = 6174
… …
现给定任意 4 位正整数，请编写程序演示到达黑洞的过程。

输入格式：
输入给出一个 (0,10
4
) 区间内的正整数 N。

输出格式：
如果 N 的 4 位数字全相等，则在一行内输出 N - N = 0000；否则将计算的每一步在一行内输出，直到 6174 作为差出现，输出格式见样例。注意每个数字按 4 位数格式输出。

输入样例 1：
6767
输出样例 1：
7766 - 6677 = 1089
9810 - 0189 = 9621
9621 - 1269 = 8352
8532 - 2358 = 6174
输入样例 2：
2222
输出样例 2：
2222 - 2222 = 0000

注意测试点：
1：输入的几位正整数相同时，应输出xxxx-xxxx=0000
2:输入的正整数不是四位数时，应进行补零操作，比如，输入10，补充后第一次计算为1000-0001=0999
3：循环操作过程中，出现被减数与减数相同的情况，要直接break循环。
代码

def list_to_number(list):  #字符列表转为数字
    num=0
    list_len=len(list)
    for i in range(list_len):
        num=num+list[i]*pow(10,list_len-i-1)
    return num

def zero_buwei(num): #补零操作
    b=num/10
    if b<1:
        return '000' + str(num)
    elif b<10:
        return '00' + str(num)
    elif b<100:
        return '0' + str(num)
    elif b < 1000:
        return str(num)

import sys
input=sys.stdin.readline()
#print(input)
input=zero_buwei(int(input))
input_list=list()
for num in input:
    if num is '\n':
        break
    else:
        input_list.append(int(num))

A=sorted(input_list,reverse=True) #降序
B=sorted(input_list,reverse=False) #升序
A_num=list_to_number(A)

B_num=list_to_number(B)
if A_num==B_num: #判断各位上的数字是否相等
    print(f'{zero_buwei(A_num)} - {zero_buwei(B_num)} = {zero_buwei(A_num - B_num)}')
else:
    while A_num - B_num !=6174:
        print(f'{zero_buwei(A_num)} - {zero_buwei(B_num)} = {zero_buwei(A_num - B_num)}')
        c_str = zero_buwei(A_num - B_num)
        C=list()
        for i in c_str:
            C.append(int(i))

        A = sorted(C, reverse=True)  # 降序
        B = sorted(C, reverse=False)  # 升序
        A_num = list_to_number(A)
        B_num = list_to_number(B)
        if A_num == B_num:  # 判断各位上的数字是否相等
            break

    print(f'{zero_buwei(A_num)} - {zero_buwei(B_num)} = {zero_buwei(A_num - B_num)}')