UVa Problem Solution: 10042 - Smith Numbers

Follow the problem's statements and you'll get it.

Code:

1. /***************************************************************************
2.  *   Copyright (C) 2008 by Liu Kaipeng                                     *
3.  *   LiuKaipeng at gmail dot com                                           *
4.  ***************************************************************************/
6. /* @JUDGE_ID 00000 10042 C++ "Smith Numbers" */
8. #include <algorithm>
9. #include <cmath>
10. #include <cstdio>
11. #include <cstring>
12. #include <deque>
13. #include <fstream>
14. #include <iostream>
15. #include <list>
16. #include <map>
17. #include <queue>
18. #include <set>
19. #include <stack>
20. #include <string>
21. #include <vector>
23. using namespace std;
24.      
25. int const max_primes = 31623;     
26. bool primes[max_primes];
28. void gen_primes()
29. {
30.   primes[0] = primes[1] = false;
31.   for (int i = 2; i < max_primes; ++i)
32.     primes[i] = true;
33.   for (int i = 2; i < max_primes; ++i) {
34.     for (; i < max_primes && !primes[i]; ++i) {}
35.     for (int j = 2; i*j < max_primes; ++j)
36.       primes[i*j] = false;
37.   } 
38. }
40. bool is_smith(int n)
41. {
42.   int s1 = 0;
43.   for (int m = n; m != 0; m /= 10) s1 += m % 10;
44.   int s2 = 0;
45.   int f = 2, mf = sqrt(n) + 1;
46.   for (; n >= f && f < mf && s2 <= s1; ++f) {
47.     if (!primes[f]) continue;
48.     if (n % f == 0) {
49.       int s = 0;
50.       for (int m = f; m != 0; m /= 10) s += m % 10;
51.       for (; n % f == 0; n /= f) s2 += s;
52.     }
53.   }
54.   if (s2 != 0 && n != 1) 
55.     for (int m = n; m != 0; m /= 10) s2 += m % 10;
56.   return s1 == s2;
57. }
58.           
59. int main(int argc, char *argv[])
60. {
61. #ifndef ONLINE_JUDGE
62.   freopen((string(argv[0]) + ".in").c_str(), "r", stdin);
63.   freopen((string(argv[0]) + ".out").c_str(), "w", stdout);
64. #endif
66.   gen_primes();
68.   int ncases;
69.   cin >> ncases;
70.   for (int n; ncases > 0 && cin >> n; --ncases) {
71.     int m = n + 1;
72.     for (; !is_smith(m); ++m) {}
73.     cout << m << '/n';
74.   }
76.   return 0;
77. }
78.