PAT 乙级 1019 数字黑洞

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

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

7766 - 6677 = 1089
9810 - 0189 = 9621
9621 - 1269 = 8352
8532 - 2358 = 6174
7641 - 1467 = 6174
... ...

现给定任意 4 位正整数，请编写程序演示到达黑洞的过程。

输入格式：

输入给出一个 (0,104) 区间内的正整数 N。

输出格式：

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

Python 

num = input()

def four(a):
    a = int(a)
    if a < 10:
        return '000' + str(a)
    elif a < 100:
        return '00' + str(a)
    elif a < 1000:
        return '0' + str(a)
    else:
        return str(a)

for i in range(len(num)):
    if not ('0' <= num[i] <= '9'):
        break
else:
    if num and 0 <= int(num) < 10000:
        num = four(num)
        if num == num[0] * len(num):
            print(four(num),'-',four(num),'= 0000')
        else:
            a, b = 0, 0
            while a-b != 6174:
                num = list(num)
                num.sort()
                a = int(''.join(num))
                b = int(''.join(num[::-1]))
                if a < b:
                    a, b = b, a
                print(a,'-',four(b),'=',end = ' ')
                print(four(str(a-b)))
                num = four(str(a-b))