PAT (Basic Level) Practice （中文）-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 <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;

int myPow(int N,int n){
    int result=1;
    for(int i=0;i<n;i++){
        result*=N;
    }
    return result;
}

int trans(int a[4]){
    int result=0;
    reverse(a, a+4);
    for(int i=0;i<4;i++){
        result+=a[i]*myPow(10,i);
    }
    reverse(a, a+4);
    return result;
}

void transBack(int a[4],int n){
    for(int i=0;i<4;i++){
        a[i]=n/(myPow(10, (3-i)));
        n=n%myPow(10, (3-i));
    }
}

void print0(int n){
    for(int i=0;i<4;i++){
        if(n/myPow(10,(3-i))==0) cout<<'0';
    }
}

int main(){
    char c[4];
    int a[4];
    for(int i=0;i<4;i++){
        cin>>c[i];
        a[i]=c[i]-'0';
    }
    
    while(1){
        
        sort(a, a+4);
        reverse(a, a+4);
        int x=trans(a);
        reverse(a, a+4);
        int y=trans(a);
        
        if(x-y==6174){
            print0(x);
            cout<<x<<" - ";
            print0(y);
            cout<<y<<" = ";
            print0(x-y);
            cout<<x-y<<endl;
            break;
        }
        else if(x-y==0){
            print0(x);
            cout<<x<<" - ";
            print0(y);
            cout<<y<<" = ";
            print0(x-y);
            cout<<endl;
            break;
        }
        else{
            print0(x);
            cout<<x<<" - ";
            print0(y);
            cout<<y<<" = ";
            print0(x-y);
            cout<<x-y<<endl;
            transBack(a, x-y);
        }
        
    }
}

修改后：

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

bool down(int a,int b){//降序
    return a>b;
}

int main(){
    int n;
    int a[4];
    scanf("%d", &n);
    a[0] = n % 10;
    a[1] = n / 10 % 10;
    a[2] = n / 100 % 10;
    a[3] = n / 1000;
    
    while(1){
            sort(a, a+4, down);
            int x = a[0] * 1000 + a[1] * 100 + a[2] * 10 + a[3];
            int y = a[3] * 1000 + a[2] * 100 + a[1] * 10 + a[0];
            if(x==y){
                printf("%04d - %04d = %04d\n",x,y,x-y);
                break;
            }
            else{
                printf("%04d - %04d = %04d\n",x,y,x-y);
                if(x-y==6174) break;
                a[0] = (x-y) % 10;
                a[1] = (x-y) / 10 % 10;
                a[2] = (x-y) / 100 % 10;
                a[3] = (x-y) / 1000;
            }
        
    }
}

一开始把几个函数都去掉了，直接写到main里面，发现还是不行，没有解决超时问题，最后把所有的输入输出都改成c风格的，然后没有用string输入一个个转换为a里面的数字，而是直接输入一个int来转换，才成功得到ac。