1019 数字黑洞 (C++ and Pyhton)

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

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

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

现给定任意 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. 输入输出，不足4位补0
2. 先输出再判断（可以直接输入6174）

#include <iostream>
#include <vector>
#include <iomanip>
#include <algorithm>

using namespace std;

bool cmp(int a, int b) {
    return a > b;
}

int main()
{
    int num, num1 = 0, num2 = 0;
    cin >> num;
    vector<int> vec(4, 0);
    while (1) {
        vec[0] = num / 1000;
        vec[1] = (num / 100) % 10;
        vec[2] = (num / 10) % 10;
        vec[3] = num % 10;

        sort(vec.begin(), vec.end(), cmp);
        num1 = vec[0] * 1000 + vec[1] * 100 + vec[2] * 10 + vec[3];

        sort(vec.begin(), vec.end());
        num2 = vec[0] * 1000 + vec[1] * 100 + vec[2] * 10 + vec[3];

        num = num1 - num2;
        printf("%04d - %04d = %04d\n", num1, num2, num1 - num2);
        
        if (num1 - num2 == 0 || num1 - num2 == 6174) {
            break;
        }
    }

    return 0;
}

 

num = int(input())
vec=[0,0,0,0]
while True:
    vec[0] = int(num / 1000)
    vec[1] = int((num / 100)) % 10
    vec[2] = int((num / 10)) % 10
    vec[3] = num % 10

    vec.sort()
    num1 = vec[0] * 1000 + vec[1] * 100 + vec[2] * 10 + vec[3]

    vec.sort(reverse=True)
    num2 = vec[0] * 1000 + vec[1] * 100 + vec[2] * 10 + vec[3]

    num = num2 - num1
    print("%04d - %04d = %04d"%(num2, num1, num), end='\n')

    if(num ==0 or num == 6174):
        break