UVa Problem Solution: 10247 - Complete Tree Labeling

Let f(k, d) denote the number of labelings of the k-ary tree of depth d. Moreover, let n(k, d) denote the number of nodes of the tree. The tree contains k subtrees, each contains n(k, d-1) nodes. There are totally  (n(k, d) - 1)! / (n(k, d-1)!)^k ways to partition the n(k, d) - 1 labels into k different subsets, each have the same size of n(k, d-1), therefore  we can draw this recurrence:
   f(k, d) = (n(k, d) - 1)! / (n(k, d-1)!)^k * f(k, d-1)
with base cases of
   f(k, 0) = 1

Code:

1. /***************************************************************************
2.  *   Copyright (C) 2008 by Liu Kaipeng                                     *
3.  *   LiuKaipeng at gmail dot com                                           *
4.  ***************************************************************************/
6. /* @JUDGE_ID 00000 10247 C++ "Complete Tree Labeling" */
8. #include <algorithm>
9. #include <cstdio>
10. #include <cstring>
11. #include <deque>
12. #include <fstream>
13. #include <iostream>
14. #include <list>
15. #include <map>
16. #include <queue>
17. #include <set>
18. #include <stack>
19. #include <string>
20. #include <vector>
21. #include "big_unsigned.hpp"
23. using namespace std;
25. /*
26.  * Let f(k, d) denote the number of labelings of the k-ary tree of depth d.
27.  * Moreover, let n(k, d) denote the number of nodes of the tree. The tree
28.  * contains k subtrees, each contains n(k, d-1) nodes. There are totally 
29.  * (n(k, d) - 1)! / (n(k, d-1)!)^k ways to partition the n(k, d) - 1 labels
30.  * into k different subsets, each have the same size of n(k, d-1), therefore 
31.  * we can draw this recurrence:
32.  *   f(k, d) = (n(k, d) - 1)! / (n(k, d-1)!)^k * f(k, d-1)
33.  * with base cases of
34.  *   f(k, 0) = 1
35.  */
36. big_unsigned numbers[22][22];
37. void gen_numbers()
38. {
39.   unsigned powers[22][22];
40.   for (int k = 1; k <= 21; ++k) {
41.     powers[k][0] = 1;
42.     for (int d = 1; d * k <= 21; ++d)
43.       powers[k][d] = powers[k][d-1] * k;
44.   }
46.   unsigned nodes[22][22];
47.   unsigned max_fac = 0;
48.   for (int k = 1; k <= 21; ++k) {
49.     nodes[k][0] = 1;
50.     for (int d = 1; d * k <= 21; ++d)
51.       if ((nodes[k][d] = nodes[k][d-1] + powers[k][d]) > max_fac)
52.         max_fac = nodes[k][d];
53.   }
55.   vector<big_unsigned> facs(max_fac);
56.   facs[0] = 1;
57.   for (unsigned i = 1; i < max_fac; ++i)
58.     facs[i] = facs[i-1] * i;
60.   for (int k = 1; k <= 21; ++k) {
61.     numbers[k][0] = 1;
62.     for (int d = 1; d * k <= 21; ++d) {
63.       numbers[k][d] = numbers[k][d-1];
64.       for (int i = 1; i != k; ++i) numbers[k][d] *= numbers[k][d-1];
65.       numbers[k][d] *= facs[nodes[k][d]-1];
66.       for (int i = 0; i < k; ++i) numbers[k][d] /= facs[nodes[k][d-1]];
67.     }
68.   }
69. }  
70.         
71. int main(int argc, char *argv[])
72. {
73. #ifndef ONLINE_JUDGE
74.   freopen((string(argv[0]) + ".in").c_str(), "r", stdin);
75.   freopen((string(argv[0]) + ".out").c_str(), "w", stdout);
76. #endif
78.   gen_numbers();
79.   for (int k, d; cin >> k >> d; ) cout << numbers[k][d] << '/n';
81.   return 0;
82. }
83.