PAT 1019 数字黑洞 (20 分)

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

#include <iostream>
#include <vector>
#include <algorithm>
#include <functional>
using namespace std;
//输出结果
void print(vector<int>& v)
{
	vector<int>::iterator it=v.begin();
	for(it;it!=v.end();it++)
	cout<<*it;
}
bool comp(int a,int b){
	return a>b;
}
//输入一个整形数据
int scan(vector<int>& c,int n)
{
//对数据拆分
	int count=0;
	int a[4]={0};
	for(int i=3;i>=0;--i)
	{
		a[i]=n%10;
		count+=a[i];
		n=n/10;	
	}
	if(count%4==0){
		return 1;
	}else{
			for(int i=0;i<4;++i)
			{
				c.push_back(a[i]);
			}
	}
}
void Sort(vector<int>& va,vector<int>& vb)
{	//递增排序va容器
	sort(va.begin(),va.end());
	vb=va;
	//递减序vb容器
	sort(vb.begin(),vb.end(),comp);
}

void addition(vector<int>& va,vector<int>& vb,vector<int>& vc)
{
	int arry[4]={0};
	int i=3;
	for(vector<int>::reverse_iterator it=va.rbegin(),ij=vb.rbegin();it!=va.rend();it++,ij++)
	{
	//高精度减法，由于之前代码写错了，看成每个数据求和，这里只能这样改成高精度减法
		if(*ij>=*it){
			arry[i--]=*ij-*it;
		}else{
			arry[i--]=*ij+10-*it;
			*(ij+1)=*(ij+1)-1;
		}
	}
	//得到的数据给vc容器
	for(int k=0;k<4;++k)
	{
		vc.push_back(arry[k]);
	}
}
//向量内容化成整形数据
int Sum(vector<int>& vc)
{
	int sum=0;
	for(vector<int>::iterator it=vc.begin();it!=vc.end();it++)
	{
		sum=sum*10+*it;	
	}
//	cout<<"sum:"<<sum;
	return sum;
	
}
int main()
{
	vector<int>va,vb,vc;
	int n,sum=0,flag=0,t;
	cin>>n;
	t=n;
	flag=scan(va,n);
	if(flag==1){
		cout<<t<<" - "<<t<<" = "<<"0000";
		return 0;
	}else{
		while(sum!=6174)
		{
			Sort(va,vb); 
			print(vb);cout<<" - ";
			print(va);cout<<" = ";
			addition(va,vb,vc);
			print(vc);cout<<endl;
			sum=Sum(vc);
			va=vc;
			//使用了容器，每次push_back都会往后增加，重复的使用vc容器，需要对它进行清空。
			vc.clear();
		}
	}
	return 0;
}