PAT乙级1019. 数字黑洞 (20)

题目

链接https://www.patest.cn/contests/pat-b-practise/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, 10000)区间内的正整数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

思路

对于给定的数，分别求出千位、百位、十位和个位，然后排序，判断最大数和最小数是否相等，如果相等话则是4个数字全部相等

然后用最大数组合-最小数组合，判断下差是不是6174，这里递归调用，直到差为6147为止

注意：如果输入的是1，则1000-0001=999是三位数(第二个测试点) 需要额外考虑，如果差是三位数，则输出0在输出差

代码

#include<iostream>
#include<string.h>
#include<algorithm>  
#include<vector>  
using namespace std;

void judge(int n)
{
	int i,a[4],num=n;
	i=0;
	memset(a,0,sizeof(a));
	while(num)//获得个位十位百位千位
	{
		a[i++]=num%10;
		num/=10;
	}
	sort(a,a+4);//这里排序需要注意 对4位数都排，防止输入的值为非四位数
	if(a[0]==a[3]) cout<<n<<" - "<<n<<" = 0000"<<endl;
	else
	{	int k ;
		k=(a[3]-a[0])*1000+(a[2]-a[1])*100+(a[1]-a[2])*10+(a[0]-a[3]);
		cout<<a[3]<<a[2]<<a[1]<<a[0]<<" - "<<a[0]<<a[1]<<a[2]<<a[3]<<" = ";
		if(k<1000) cout<<"0";//判断差与1000的关系
		cout<<k<<endl;
		if(k!=6174)	
			judge(k);//递归
	}
}

int main()
{
	//freopen("in.txt" , "r" , stdin) ;
	int n;
	cin>>n;
	judge(n);	
	return 0;
}