PAT 乙級 1019. 數字黑洞 (C++版)

給定任一個各位數字不完全相同的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 <stdio.h>
using namespace std;
int ascend,descend;
void getNumber(int n){
    int a[4];
    for(int i = 0;i < 4; ++i){
        a[i] = n % 10;
        n = n / 10;
    }
    sort(a,a+4);
    descend = a[3]*1000 + a[2]*100 + a[1]*10 + a[0];
    ascend = a[0]*1000 + a[1]*100 + a[2]*10 + a[3];
}
int main(){
    int n;
    cin >> n ;
    do{
        getNumber(n);
        n = descend - ascend;
        printf("%04d - %04d = %04d\n",descend,ascend,n);
        if(ascend == descend){
            break;
        }
        //6174是一個測試點
    }while( n != 6174);
    return 0;
}