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<sstream>
#include<string>
using namespace std;
string num2str(int i)//数字转字符串 
{
	stringstream ss;
	ss<<i;
	return ss.str();
}
int str2num(string s)//字符串转数字 
{
	int num;
	stringstream ss(s);
	ss>>num;
	return num;
}
int main()
{
	string s;//输入的第一个数
	cin>>s;
	if(s.length()<4)//可能不够4位 
	{
		if(s.length()==1)
		{
			s=s+"000";
		}
		else if(s.length()==2)
		{
			s=s+"00";
		}
		else if(s.length()==3)
		{
			s=s+"0";
		 } 
	}
	string s1;
	string s2;
	int A;//值
	A=0;
	while(A!=6174)
	{
		sort(s.begin(),s.end());//得到升序序列 
		s1=s;
		reverse(s.begin(),s.end());//得到降序序列
		s2=s;
		A=str2num(s2)-str2num(s1);//转成数字 
		if(A==0)
		{
			cout<<s2<<" - "<<s1<<" = 0000"<<endl;break;
		}
		else{
			cout<<s2<<" - "<<s1<<" = "<<A<<endl;
			s=num2str(A);//赋值再次判断循环 
		}
	}
}

结果只得了18分，不知错哪里了，在牛客网是满分