PAT乙级 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

注意事项

第一个输入的数字是6174时，也要输出

代码

#include<stdio.h>
#include<math.h>
void func(int N)
{
    if (N == 6174)return;

    int a[4] = { 0 }, b[4] = { 0 }, size = 0, i = 0, j;
    while (N)
    {
        a[size] = N % 10;
        b[size] = a[size];
        N /= 10;
        size++;
    }
    int flag = 0;
    for (i = 0; i < 3; ++i)
    {
        if (a[i] != a[i + 1])
            flag = 1;
    }

    for ( i= 0; i < 4; ++i)
    {
        for (j = i + 1; j < 4; ++j)
        {
            if (a[j] > a[i])
            {
                int temp1 = a[i];
                a[i] = a[j];
                a[j] = temp1;
            }
            if (b[j] < b[i])
            {
                int temp2 = b[i];
                b[i] = b[j];
                b[j] = temp2;
            }
        }
    }
    int result1 = 0, result2 = 0;
    for (i = 0; i < 4; ++i)
    {
        result1 = result1 + a[i] * pow(10, 3 - i);
        result2 = result2 + b[i] * pow(10, 3 - i);
    }
    printf("%04d - %04d = %04d\n", result1, result2, result1 - result2);
    if (flag == 0)return;
    func(result1 - result2);
}


int main()
{
  
    int N;
    scanf("%d", &N);
    if (N == 6174)
        printf("7641 - 1467 = 6174\n");
    func(N);
    return 0;
}