PAT 1019 数字黑洞 python 有超时

1019 数字黑洞 (20)（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, 10000)区间内的正整数N。

输出格式：

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

n = input()
if len(n) < 4:
    n = '0' * (4 - len(str(n))) + n
n = list(n)
m = sorted(n, reverse=True)
p = sorted(n, reverse=False)
s1 = ''
s1 = s1.join(m)
s2 = ''
s2 = s2.join(p)
k = int(s1) - int(s2)
if k == 0:
    print(s1+' - '+s2+' = '+"0000")
else:
    print(s1+' - '+s2+' = '+str(k))
    while k != 6174:
        k = list(str(k))
        d = sorted(k, reverse=True)
        c = sorted(k, reverse=False)
        s3 = ''
        s3 = s3.join(d)
        s4 = ''
        s4 = s4.join(c)
        k = int(s3) - int(s4)
        print(s3+' - '+s4+' = '+str(k))

我的代码有一个测试点超时，做这个题的时候忘记判断输入的字符是不是4位数，所以一定要注意，请大神多指教