PAT-乙-1019 数字黑洞

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 <string.h>
#include <cstdio>
#include <iomanip>
#include <algorithm>
#include<sstream>
using namespace std;
bool cmp(char s1,char s2){
    return s1>s2;
}
string sub(string s1,string s2){
    string s="1234";
    stringstream ss;
    int a,b;
    ss<<s1;
    ss>>a;
    ss.clear();
    ss<<s2;
    ss>>b;
    int c=a-b;
    ss.clear();
    ss<<c;
    ss>>s;
    while(s.size()!=4)
        s.insert(0,"0");
    return s;
}
int main(){
    string s,s1,s2;
    cin>>s;
    while(s.size()!=4)
        s.insert(0,"0");
    if(s[0]==s[1]&&s[2]==s[3]&&s[1]==s[2]){
            cout<<s<<" - "<<s<<" = "<<"0000";
    }
    else while(s!="6174"||s1==s2){
        s1=s;
        s2=s;
        sort(s1.begin(),s1.end(),cmp);
        sort(s2.begin(),s2.end());
        s=sub(s1,s2);
        cout<<s1<<" - "<<s2<<" = "<<s;
        cout<<endl;
    }
}

注解
1、有一个坑点，输入本身是6174，应该输出7641 - 1467 = 6174！！！！
2、本题学到的语法：
（1）int与string的互相转换，利用stringstream作桥，stringstream每次用完必须clear才能再次使用
（2）string排序可用sort，cmp函数返回值为bool 且输入为char