PAT 1019. 数字黑洞

原题链接：http://pat.zju.edu.cn/contests/pat-b-practise/1019

给定任一个各位数字不完全相同的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 <stdio.h>
#include <vector>
#include <algorithm>
using namespace std;
const int Kaprekar = 6174;
int main()
{
  int m;
  int a;
  vector<int> digits;
  int big_data, small_data;
  while(scanf("%d", &m)!=EOF)
  {
    do{
      digits.clear();
      while(m!=0)
      {
        a = m%10;
        digits.push_back(a);
        m = m/10;
      }
      digits.resize(4,0);//防止在计算过程中出现不足4个数字的情况
      sort(digits.begin(), digits.end());
      big_data = small_data = 0;
      int k = 1;
      for(int i=0; i<4; ++i)
      {
        big_data += (digits[i] * k);
        k *= 10;
      }
      if((digits[0]==digits[1])&&(digits[1]==digits[2])&&(digits[2]==digits[3]))
      {
        printf("%04d - %04d = 0000\n", big_data, big_data);
        break;
      }
      k = 1;
      for(int i=3; i>=0; --i)
      {
        small_data += (digits[i] * k);
        k *= 10;
      }
      m = big_data - small_data;
      printf("%04d - %04d = %04d\n", big_data, small_data, m);
    }while(m!=Kaprekar);
  }
  return 0;
}