蓝桥杯算法提高VIP-数字黑洞 （C++代码）

#include<iostream>
#include<algorithm>
using namespace std;
int compare(int a,int b){
    return a>b;
}
int main(){
    int a[4],b[4],flag,n=0;
    cin>>flag;
    for(int i=3;i>=0;i--){
        a[i]=flag%10;
        flag=flag/10;
    }
    while(1){
        for(int i=0;i<4;i++)
            b[i]=a[i];
        sort(b,b+4,compare);
        sort(a,a+4);
        for(int i=3;i>=0;i--){
            if((b[i]-a[i])<0){
                b[i]=b[i]+10;
                b[i-1]=b[i-1]-1;
            }
             a[i]=b[i]-a[i];
        }
        n++;
        if(a[0]==6&&a[1]==1&&a[2]==7&&a[3]==4){
            cout<<n<<endl;
            break;}
    }
    return 0;
}