【PAT】甲级A1069 The Black Hole of Numbers (20 分)/乙级B1019 数字黑洞 (20 分)

PAT中A1069和B1019都是数字黑洞问题，这里放上中文版本，方便阅读。

给定任一个各位数字不完全相同的 4 位正整数，如果我们先把 4 个数字按非递增排序，再按非递减排序，然后用第 1 个数字减第 2 个数字，将得到一个新的数字。一直重复这样做，我们很快会停在有“数字黑洞”之称的 6174，这个神奇的数字也叫 Kaprekar 常数。

例如，我们从6767开始，将得到

7766 - 6677 = 1089
9810 - 0189 = 9621
9621 - 1269 = 8352
8532 - 2358 = 6174
7641 - 1467 = 6174
… …
现给定任意 4 位正整数，请编写程序演示到达黑洞的过程。

输入格式：

输入给出一个 (0,1e4 ) 区间内的正整数 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<iostream>
#include<algorithm>
#include<iomanip>
using namespace std;

int Minus(int* a,int* b){
	int p=0,q=0;
	int i=0;
	while(i<4){
		p=a[i]+p*10;
		q=b[i]+q*10;
		i++;
	}
	return p-q;
}
bool cmp(int a,int b){
	return a>b;
}
int main(){
	int a;
	cin>>a;
	do{
		int i=0;
		int num1[4]={0},num2[4]={0};//不能省略！否则若输入不足4位数字时会出错 
		while(a){
			num1[i]=num2[i]=a%10;
			i++;
			a/=10;
		}
		sort(num1,num1+4,cmp);//递减排序
		sort(num2,num2+4);//递增排序
		for(i=0;i<4;i++)
			cout<<num1[i]; 
		cout<<" - ";
		for(i=0;i<4;i++)
			cout<<num2[i]; 
		cout<<" = ";
		a=Minus(num1,num2);
		cout<<setfill('0')<<setw(4)<<a<<endl;//输出4位 
	}while(a!=0&&a!=6174);
	return 0;
}

需要注意的是，刚开始我提交遇到这样的问题——测试点234未通过：

后来发现问题在于我的代码只能够处理输入为4位数的情况，若输入123位数则会出错，在代码中为数组赋初值即解决问题。