PAT-B 1019. ディジタルブラックホール(PAT 1069.The Black Hole of Numbers)

1.N-N=0の特判
2.最初の入力は6174の特判(私の233のような実装方法が必要)
コード:

#include <cstdio>
#include <algorithm>
#include <string>
#include <sstream>

using namespace std;

int n;

bool mycmp(const char a, const char b)
{
	return a > b;
}

bool same_numbers(int n)
{
	for (int i = 0; i < 10; ++ i)
	{
		if (n == 1111*i)
		{
			return true;
		}
	}
	return false;
}

void print_steps(int b)
{
	if (b == 6174)
	{
		printf("7641 - 1467 = 6174
");
		return ;
	}

	string digit;
	stringstream ss;
	int a;
	while (b != 6174)
	{
		// get a
		digit.clear();
		ss.clear();
		ss << b;
		ss >> digit;
		while (digit.size() < 4)
		{
			digit.push_back('0');
		}
		sort(digit.begin(), digit.end());
		ss.clear();
		ss << digit;
		ss >> b;
		sort(digit.begin(), digit.end(), mycmp);
		ss.clear();
		ss << digit;
		ss >> a;
		printf("%04d - %04d = %04d
", a, b, a-b);
		b = a - b;
	}
}

int main()
{
	scanf("%d", &n);

	if ( same_numbers(n) )
	{
		printf("%04d - %04d = 0000
", n, n);
	} else
	{
		print_steps(n);
	}

	return 0;
}