PAT 1019. 数字黑洞 (20)

交第一遍的时候被测试点2卡住了，后来才知道还有关于位数小于4位的数，存在数组里的问题没有处理。

#include <iostream>
#include <algorithm>
using namespace std;

bool cmp(int x, int y) {
	return x > y;
}

void deal() {
	int n;
	cin >> n;
	int x = n, i = 0, y = 0;
	if (x == 6174)
	{
		printf("7641 - 1467 = 6174\n");
		return;
	}

	int a[4] = { 0 };
	while (x - y != 6174) {		
		i = 0;
		while (x != 0) {
			a[i++] = x % 10;
			x /= 10;
		}
		for (i = i; i < 4; i++)
			a[i] = 0;
		//这里的测试点2是当后面的n小于4位 比如输入1 1000-0001= 0999 这里只存了3位 还有一位默认是之前的1 所以加了一个循环 把没有存的数初始化为0

		sort(a, a + 4, cmp);
		for (i = 0; i < 4; i++)
		{
			cout << a[i];
			x = x * 10 + a[i];
		}
		cout << " - ";

		sort(a, a + 4);
		for (i = 0; i < 4; i++)
		{
			cout << a[i];
			y = y * 10 + a[i];
		}

		printf(" = %04d\n", x - y);
		//这里的%04d表示没有四位用0补齐，很方便

		x = x - y;
		y = 0;
		if (x - y == 0)
			break;
	}
}
int main() {
	deal();
	return 0;
}