PAT 乙级 数字黑洞

题目

给定任一个各位数字不完全相同的 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

分析

这是一道高精度 + 递归 的题

AC代码

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

using namespace std;

int flag;
int ans[4] = {6, 1, 7, 4};

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

void sub (vector<int> a)
{
	//递归结束条件 
	int cnt = 0;
	for(int i = 0 ; i <= 3 ; i ++ )
	{
	if(a[i] == ans[i]) cnt ++ ;
	}
	if(cnt == 4) return ;
	
	while(a.size() < 4) a.push_back(0);
	 
	sort(a.begin(), a.end(), cmp); 
	vector<int> b(a);
	sort(a.begin(), a.end());  
	vector<int> c(a);
	
	while(!a.empty())
        	a.pop_back();

	int t = 0;
	for(int i = 3 ; i >= 0 ; i -- )
	{
	    t = b[i] - t;
	    if(i >= 0) t -= c[i];
	    a.push_back((t + 10) % 10);
	    t = t < 0 ? 1 : 0 ; 
	}
	
	reverse(a.begin(), a.end());
	
	for(int i = 0 ; i <= 3 ; i ++ ) printf("%d", b[i]);
	printf(" - ");
	
	for(int i = 0 ; i <= 3 ; i ++ ) printf("%d", c[i]);
	printf(" = ");
	
	for(int i = 0 ; i <= 3 ; i ++ ) printf("%d", a[i]);
	printf("\n");
	
	sub(a);
}

int main ()
{
	 string a;
	 cin>>a;
	 
	 for(int i = 0 ; i < a.size() ; i ++ )
	 {
	  if(a[i] != a[1]) flag = 1;
	 }
	 if(!flag)
	 {
	  cout<<a<<" - "<<a<<" = "<<"0000"<<"\n";
	  return 0; 
	 }  
	 
	 //形如 -> vector<int> a(3) vector数组所有元素会被自动初始化为0,长度为3 
	 vector<int> A;  
	 for(int i = 0 ; i < a.size() ; i ++ ) A.push_back(a[i] - '0');
	    
	 reverse(A.begin(), A.end());
    
     sub(A);
    
    return 0;
}