1019 数字黑洞 (20 分) - 解决测试点2

PAT乙级真题

题目链接：

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

引入：

给定任一个各位数字不完全相同的 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⁴) 区间内的正整数 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位数，判断各位数是否完全相同
2 非递增 递减和相等 为A
非递减 递增和相等 为B
3 B - A 判断是否等于6174，不等于就继续
解题思路
① 用字符串存储整数，并用s.insert(0, 4 - s.length(), '0');将缺位用0补全，再分别判断每位是否相等
② 用sort()函数对整数字符串进行排序，存放到两个新的字符串中
③ 用stoi()函数将字符串转为数字并相减得到结果，定义为re
④ 将re转为字符串并再次补全缺位，进行重复操作，直到re为6174停止循环

测试点
测试点0 不过一般是运算超时，输入测试组 0000 看问题出在哪
测试点2 在多次尝试后，我发现输出的整数要都为字符串形式，输出整型过不了

#include <iostream>
#include <algorithm>
using namespace std;
bool cmp(char a, char b){
	return a > b;
}
int main(){
	string s, a, b;
	int re = 0;
	cin >> s;
    s.insert(0, 4 - s.length(), '0');		//从s的第0个位置开始，对4 - s.length()的长度插入字符'0’
	if(s[0] == s[1] && s[1] == s[2] && s[2] == s[3]){	//判断是否不完全相同
		cout << s << " - " << s << " = 0000";  
		return 0;
	}

	do{
		a = s;
		b = s;
		sort(a.begin(), a.end(), cmp);		//非递增
		sort(b.begin(), b.end());			//非递减
		re = stoi(a) - stoi(b);				
		s = to_string(re);					//将re转为字符串
		s.insert(0, 4 - s.length(), '0');	
		cout << a << " - " << b << " = " << s << endl;
	}while(re != 6174);	
	return 0;		
}

悄咪咪表白柳婼大佬