PAT (Basic Level) Practice （中文）1019 数字黑洞 python

出处：
https://pintia.cn/problem-sets/994805260223102976/problems/994805302786899968

1019 数字黑洞 (20 分) 给定任一个各位数字不完全相同的 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

这个题主要注意三点：
第一：

如果 N 的 4 位数字全相等，则在一行内输出 N - N = 0000

第二：

每个数字按 4 位数格式输出。

第三：

输入数字为6174时的输出。

代码：

n = input()

if n.count(n[0]) == 4:  #如果 N 的 4 位数字全相等，则在一行内输出 N - N = 0000
    print(n,'-',n,'=','0000')
elif n == '6174':   #如果输入是6174，输出是7641 - 1467 = 6174
    print('7641 - 1467 = 6174')
else:
    while n != '6174':  #其他情况按题目要求输出
        while len(n) != 4:
            n = '0' + n
        a = ''.join(sorted(n,reverse = True))
        b = ''.join(sorted(n))
        n = str(int(a) - int(b))
        while len(n) != 4:
            n = '0' + n
        print(a,'-',b,'=',n)

之前的错误都是因为没有注意到上面的这三点：
错误代码：

n = int(input())
e = n

def right_sort(num):
    num = str(num)
    a = []
    for i in range(len(num)):
        a.append(int(num[i]))
    a.sort()
    res = ''
    for i in range(len(num)):
        res += str(a[i])
    res = int(res)
    return res

def negative_sort(num):
    num = str(num)
    a = []
    for i in range(len(num)):
        a.append(int(num[i]))
    a.sort(key=None,reverse=True)
    res = ''
    for i in range(len(num)):
        res += str(a[i])
    res = int(res)
    return res
if str(e).count(str(e)[0]) == 4:
    print(e,'-',e,'=','0000')
else:
    while (e != 6174):
        c = negative_sort(e)
        d= right_sort(e)
        blackhole = e = c - d
        while len(str(c)) != 4:
            c = '0' + str(c)
        while len(str(d)) != 4:
            d = '0' + str(d)
        while len(str(e)) != 4:
            e = '0' + str(e)

        print(c,'-', d,'=',e)

最开始写的，比较复杂而且全是错误。
错误代码2:

n = input()

if n.count(n[0]) == 4:
    print(n,'-',n,'=','0000')
else:
    while n != '6174':
        while len(n) != 4:
            n = '0' + n
        a = ''.join(sorted(n,reverse = True))
        b = ''.join(sorted(n))
        n = str(int(a) - int(b))
        while len(n) != 4:
            n = '0' + n
        print(a,'-',b,'=',n)

这个考虑了如果 N 的 4 位数字全相等，则在一行内输出 N - N = 0000和每个数字按 4 位数格式输出。
但是有一个测试没通过。
后来自己测试输入6714:

6174

Process finished with exit code 0

发现没有输出，问题应该是出在这里。
一个简单的方法就是加个判断：

elif n == '6174':   #如果输入是6174，输出是7641 - 1467 = 6174
    print('7641 - 1467 = 6174')

就可以了。