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

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
int getNum(vector<int> nums)
{
    int result=0,mut=1;
    for(vector<int>::iterator it=nums.end();it!=nums.begin();it--)
    {
        result=result+*(it-1)*mut;
        mut=mut*10;
    }
    return result;
}
int program(int num)
{
    vector<int> nums;
    for(int i=0;i<4;i++)
    {
        int tmp=num%10;
        nums.push_back(tmp);
        num=num/10;
    }
    sort(nums.begin(),nums.end(),greater<int>());
    int max=getNum(nums);
    sort(nums.begin(),nums.end(),less<int>());
    int min=getNum(nums);
    int tmp=max-min;
    cout<<max<<"-"<<min<<"="<<tmp<<endl;
    return tmp;
}
int main()
{
    cout<<"please input a num."<<endl;
    int num;
    cin>>num;
    while(num<1000||num>9999)
    {
        cout<<"Input is wrong,please input a num again."<<endl;
        cin>>num;
    }
    do{
        num=program(num);
    }while((num!=6174)&&(num!=0));//这里需要使用do...while，要不然直接输入6174就什么也不输出了。

}