PTA | 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,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

部分正确代码

#include<bits/stdc++.h>
using namespace std;

bool f1(int n) {
	int t = n % 10;
	n /= 10;
	while (n) {
		if (t != n % 10)return false;
		n /= 10;
	}
	return true;
}

int main() {
	int n;
	cin >> n;
	if (f1(n)) {
		printf("%04d - %04d = %04d\n", n, n, n - n);
		return 0;
	}
	string s = to_string(n);
	sort(s.begin(), s.end());
	n = stoi(s);
	reverse(s.begin(),s.end());
	int t= stoi(s);
	while (t - n != 6174) {
		printf("%04d - %04d = %04d\n", t, n, t - n);
		s = to_string(t - n);
		sort(s.begin(), s.end());
		n = stoi(s);
		reverse(s.begin(), s.end());
		t = stoi(s);
	}
	//cout << t << " - " << n << " = " << t - n << endl;
	printf("%04d - %04d = %04d\n", t, n, t - n);

	return 0;
}
//没考虑清楚时间复杂度

考虑了学姐的代码以后修改

分析：有一个测试用例注意点，如果当输入N值为6174的时候，依旧要进行下面的步骤，直到差值为6174才可以～所以用do while语句，无论是什么值总是要执行一遍while语句，直到遇到差值是0000或6174～

s.insert(0, 4 – s.length(), ‘0’);用来给不足4位的时候前面补0～

#include<bits/stdc++.h>
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 result = stoi(a) - stoi(b);
		s = to_string(result);
		s.insert(0, 4 - s.length(), '0');
		cout << a << " - " << b << " = " << s << endl;
	} while (s != "6174" && s != "0000");
	return 0;
}

Ps.感觉自己写代码的时候还是有很多不足