PAT乙级1019 数字黑洞（C语言）

题目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 位数格式输出。

输入样例 1：

6767

输出样例 1：

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

输入样例 2：

2222

输出样例 2：

2222 - 2222 = 0000

思路

用字符串存储输入数字，定义两个变量存储并分类排序，转换成整形（stoi()）后相减，因退出循环需要比较条件，故又转换回string类型（to_string()），其中退出条件包括“6174”“和0000”。

代码

#include <iostream>
#include <algorithm>
using namespace std;

bool cmp(char a, char b) {//单个字符
    return a > b;//降序,大的放前面
}
int main(){
    string s;
    cin >> s;
    do {
        string a = s, b = s;//更新
        sort(a.begin(), a.end(), cmp);//降序
        sort(b.begin(), b.end());//升序
        int num = stoi(a) - stoi(b);//因为需要相减，转换成十进制
        if(num == 0){
            cout << a <<" - "<< b <<" = 0000" <<endl;
            break;
        }
        s = to_string(num);//转换回string
        cout << a <<" - "<< b <<" = "<< s << endl;
    }while(s != "0000" && s != "6174");
    return 0;
}

问题

以上只考虑了相减结果不存在0和全为0的情况，没有考虑4个数中存在1~3个0的情况。

优化思路

若在两数相减后的结果小于4位数，就不会有最高位的0存在，所以需要对结果“补0”。用insert()函数。

代码

#include <iostream>
#include <algorithm>
using namespace std;

bool cmp(char a, char b) {//单个字符
    return a > b;//降序,大的放前面
}
int main(){
    string s;
    cin >> s;
    s.insert(0, 4-s.length(), '0');//这一步想不到，参考柳神的
    do {
        string a = s, b = s;//更新
        sort(a.begin(), a.end(), cmp);//降序
        sort(b.begin(), b.end());//升序
        int num = stoi(a) - stoi(b);//因为需要相减，转换成十进制
        s = to_string(num);//转换回string
        s.insert(0, 4-s.length(), '0');
        cout << a <<" - "<< b <<" = "<< s << endl;
    }while(s != "0000" && s != "6174");
    return 0;
}

总结

进一步熟悉了insert()的用法str.insert(1,s)表示在str下标为1的字符前插入s，有一点想不明白，输入的四位数字就算有0，因为是字符串所以应该不会消失，为什么还要在相减运算前“补0”。