Hex Artifact Content
Not logged in

Artifact cfe4f511e056164f5955cfb6376c7da16e83dba2:


0000: 23 69 6e 63 6c 75 64 65 20 3c 69 6f 73 74 72 65  #include <iostre
0010: 61 6d 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 73  am>..#include <s
0020: 73 74 72 65 61 6d 3e 0d 0a 23 69 6e 63 6c 75 64  stream>..#includ
0030: 65 20 3c 69 6f 6d 61 6e 69 70 3e 0d 0a 23 69 6e  e <iomanip>..#in
0040: 63 6c 75 64 65 20 3c 76 65 63 74 6f 72 3e 0d 0a  clude <vector>..
0050: 23 69 6e 63 6c 75 64 65 20 3c 73 74 72 69 6e 67  #include <string
0060: 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 6d 61 70  >..#include <map
0070: 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 73 65 74  >..#include <set
0080: 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 61 6c 67  >..#include <alg
0090: 6f 72 69 74 68 6d 3e 0d 0a 23 69 6e 63 6c 75 64  orithm>..#includ
00a0: 65 20 3c 6e 75 6d 65 72 69 63 3e 0d 0a 23 69 6e  e <numeric>..#in
00b0: 63 6c 75 64 65 20 3c 69 74 65 72 61 74 6f 72 3e  clude <iterator>
00c0: 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 66 75 6e 63  ..#include <func
00d0: 74 69 6f 6e 61 6c 3e 0d 0a 23 69 6e 63 6c 75 64  tional>..#includ
00e0: 65 20 3c 63 6f 6d 70 6c 65 78 3e 0d 0a 23 69 6e  e <complex>..#in
00f0: 63 6c 75 64 65 20 3c 71 75 65 75 65 3e 0d 0a 23  clude <queue>..#
0100: 69 6e 63 6c 75 64 65 20 3c 73 74 61 63 6b 3e 0d  include <stack>.
0110: 0a 23 69 6e 63 6c 75 64 65 20 3c 63 6d 61 74 68  .#include <cmath
0120: 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 3c 63 61 73  >..#include <cas
0130: 73 65 72 74 3e 0d 0a 23 69 6e 63 6c 75 64 65 20  sert>..#include 
0140: 3c 74 75 70 6c 65 3e 0d 0a 75 73 69 6e 67 20 6e  <tuple>..using n
0150: 61 6d 65 73 70 61 63 65 20 73 74 64 3b 0d 0a 74  amespace std;..t
0160: 79 70 65 64 65 66 20 6c 6f 6e 67 20 6c 6f 6e 67  ypedef long long
0170: 20 4c 4c 3b 0d 0a 74 79 70 65 64 65 66 20 63 6f   LL;..typedef co
0180: 6d 70 6c 65 78 3c 64 6f 75 62 6c 65 3e 20 43 4d  mplex<double> CM
0190: 50 3b 0d 0a 0d 0a 73 74 72 75 63 74 20 55 6e 69  P;....struct Uni
01a0: 6f 6e 46 69 6e 64 0d 0a 7b 0d 0a 09 76 65 63 74  onFind..{...vect
01b0: 6f 72 3c 69 6e 74 3e 20 75 66 2c 20 73 7a 3b 0d  or<int> uf, sz;.
01c0: 0a 09 69 6e 74 20 6e 63 3b 0d 0a 0d 0a 09 55 6e  ..int nc;.....Un
01d0: 69 6f 6e 46 69 6e 64 28 69 6e 74 20 4e 29 20 3a  ionFind(int N) :
01e0: 20 75 66 28 4e 29 2c 20 73 7a 28 4e 2c 20 31 29   uf(N), sz(N, 1)
01f0: 2c 20 6e 63 28 4e 29 0d 0a 09 7b 0d 0a 09 09 66  , nc(N)...{....f
0200: 6f 72 20 28 69 6e 74 20 69 20 3d 20 30 3b 20 69  or (int i = 0; i
0210: 3c 4e 3b 20 2b 2b 69 29 20 75 66 5b 69 5d 20 3d  <N; ++i) uf[i] =
0220: 20 69 3b 0d 0a 09 7d 0d 0a 09 69 6e 74 20 73 69   i;...}...int si
0230: 7a 65 28 29 0d 0a 09 7b 0d 0a 09 09 72 65 74 75  ze()...{....retu
0240: 72 6e 20 6e 63 3b 0d 0a 09 7d 0d 0a 09 69 6e 74  rn nc;...}...int
0250: 20 73 69 7a 65 28 69 6e 74 20 61 29 0d 0a 09 7b   size(int a)...{
0260: 0d 0a 09 09 72 65 74 75 72 6e 20 73 7a 5b 46 69  ....return sz[Fi
0270: 6e 64 28 61 29 5d 3b 0d 0a 09 7d 0d 0a 09 69 6e  nd(a)];...}...in
0280: 74 20 46 69 6e 64 28 69 6e 74 20 61 29 0d 0a 09  t Find(int a)...
0290: 7b 0d 0a 09 09 72 65 74 75 72 6e 20 75 66 5b 61  {....return uf[a
02a0: 5d 20 3d 3d 20 61 20 3f 20 61 20 3a 20 75 66 5b  ] == a ? a : uf[
02b0: 61 5d 20 3d 20 46 69 6e 64 28 75 66 5b 61 5d 29  a] = Find(uf[a])
02c0: 3b 0d 0a 09 7d 0d 0a 09 2f 2f 20 53 65 6d 69 2d  ;...}...// Semi-
02d0: 63 6f 6d 70 72 65 73 73 69 6f 6e 20 77 2e 6f 2e  compression w.o.
02e0: 20 72 65 63 75 72 73 69 6f 6e 2e 0d 0a 09 2f 2f   recursion....//
02f0: 7b 20 77 68 69 6c 65 28 75 66 5b 61 5d 20 21 3d  { while(uf[a] !=
0300: 20 61 29 20 61 3d 75 66 5b 61 5d 3d 75 66 5b 75   a) a=uf[a]=uf[u
0310: 66 5b 61 5d 5d 3b 20 72 65 74 75 72 6e 20 61 3b  f[a]]; return a;
0320: 20 7d 0d 0a 09 62 6f 6f 6c 20 55 6e 69 6f 6e 28   }...bool Union(
0330: 69 6e 74 20 61 2c 20 69 6e 74 20 62 29 0d 0a 09  int a, int b)...
0340: 7b 0d 0a 09 09 61 20 3d 20 46 69 6e 64 28 61 29  {....a = Find(a)
0350: 3b 0d 0a 09 09 62 20 3d 20 46 69 6e 64 28 62 29  ;....b = Find(b)
0360: 3b 0d 0a 09 09 69 66 20 28 61 20 21 3d 20 62 29  ;....if (a != b)
0370: 0d 0a 09 09 7b 0d 0a 09 09 09 69 66 20 28 73 7a  ....{.....if (sz
0380: 5b 61 5d 20 3e 3d 20 73 7a 5b 62 5d 29 20 73 77  [a] >= sz[b]) sw
0390: 61 70 28 61 2c 20 62 29 3b 0d 0a 09 09 09 75 66  ap(a, b);.....uf
03a0: 5b 61 5d 20 3d 20 62 3b 0d 0a 09 09 09 73 7a 5b  [a] = b;.....sz[
03b0: 62 5d 20 2b 3d 20 73 7a 5b 61 5d 3b 0d 0a 09 09  b] += sz[a];....
03c0: 09 2d 2d 6e 63 3b 0d 0a 09 09 7d 0d 0a 09 09 72  .--nc;....}....r
03d0: 65 74 75 72 6e 20 28 61 20 21 3d 20 62 29 3b 0d  eturn (a != b);.
03e0: 0a 09 7d 0d 0a 7d 3b 0d 0a 0d 0a 73 74 61 74 69  ..}..};....stati
03f0: 63 20 63 6f 6e 73 74 20 75 6e 73 69 67 6e 65 64  c const unsigned
0400: 20 4d 4f 44 56 41 4c 20 3d 20 31 30 30 30 30 30   MODVAL = 100000
0410: 30 30 30 37 3b 0d 0a 0d 0a 63 6c 61 73 73 20 44  0007;....class D
0420: 65 6e 67 6b 6c 65 6b 47 61 6e 65 73 68 41 6e 64  engklekGaneshAnd
0430: 54 72 65 65 20 7b 20 70 75 62 6c 69 63 3a 0d 0a  Tree { public:..
0440: 09 69 6e 74 20 67 65 74 43 6f 75 6e 74 28 73 74  .int getCount(st
0450: 72 69 6e 67 20 6c 61 62 65 6c 73 2c 20 76 65 63  ring labels, vec
0460: 74 6f 72 20 3c 69 6e 74 3e 20 70 61 72 65 6e 74  tor <int> parent
0470: 73 29 0d 0a 09 7b 0d 0a 09 09 63 6f 6e 73 74 20  s)...{....const 
0480: 69 6e 74 20 4e 20 3d 20 6c 61 62 65 6c 73 2e 73  int N = labels.s
0490: 69 7a 65 28 29 3b 0d 0a 0d 0a 09 09 76 65 63 74  ize();......vect
04a0: 6f 72 3c 76 65 63 74 6f 72 3c 69 6e 74 3e 3e 20  or<vector<int>> 
04b0: 63 68 69 6c 64 72 65 6e 28 4e 29 3b 0d 0a 09 09  children(N);....
04c0: 55 6e 69 6f 6e 46 69 6e 64 20 75 66 28 4e 29 3b  UnionFind uf(N);
04d0: 0d 0a 09 09 66 6f 72 20 28 69 6e 74 20 69 20 3d  ....for (int i =
04e0: 20 31 3b 20 69 20 3c 20 4e 3b 20 2b 2b 69 29 20   1; i < N; ++i) 
04f0: 7b 0d 0a 09 09 09 69 6e 74 20 75 20 3d 20 70 61  {.....int u = pa
0500: 72 65 6e 74 73 5b 69 2d 31 5d 2c 20 76 20 3d 20  rents[i-1], v = 
0510: 69 3b 0d 0a 09 09 09 63 68 69 6c 64 72 65 6e 5b  i;.....children[
0520: 75 5d 2e 70 75 73 68 5f 62 61 63 6b 28 76 29 3b  u].push_back(v);
0530: 0d 0a 09 09 09 69 66 20 28 6c 61 62 65 6c 73 5b  .....if (labels[
0540: 75 5d 20 3d 3d 20 6c 61 62 65 6c 73 5b 76 5d 29  u] == labels[v])
0550: 0d 0a 09 09 09 09 75 66 2e 55 6e 69 6f 6e 28 75  ......uf.Union(u
0560: 2c 20 76 29 3b 0d 0a 09 09 7d 0d 0a 0d 0a 09 09  , v);....}......
0570: 76 65 63 74 6f 72 3c 69 6e 74 3e 20 6c 65 76 65  vector<int> leve
0580: 6c 28 4e 29 3b 0d 0a 09 09 7b 0d 0a 09 09 09 66  l(N);....{.....f
0590: 75 6e 63 74 69 6f 6e 3c 76 6f 69 64 28 69 6e 74  unction<void(int
05a0: 2c 20 69 6e 74 29 3e 20 72 65 63 3b 20 72 65 63  , int)> rec; rec
05b0: 20 3d 20 5b 26 5d 28 69 6e 74 20 76 2c 20 69 6e   = [&](int v, in
05c0: 74 20 6c 76 29 20 7b 0d 0a 09 09 09 09 6c 65 76  t lv) {......lev
05d0: 65 6c 5b 76 5d 20 3d 20 6c 76 3b 0d 0a 09 09 09  el[v] = lv;.....
05e0: 09 66 6f 72 20 28 69 6e 74 20 75 20 3a 20 63 68  .for (int u : ch
05f0: 69 6c 64 72 65 6e 5b 76 5d 29 0d 0a 09 09 09 09  ildren[v])......
0600: 09 72 65 63 28 75 2c 20 6c 76 20 2b 20 31 29 3b  .rec(u, lv + 1);
0610: 0d 0a 09 09 09 7d 3b 0d 0a 09 09 09 72 65 63 28  .....};.....rec(
0620: 30 2c 20 30 29 3b 0d 0a 09 09 7d 0d 0a 0d 0a 09  0, 0);....}.....
0630: 09 6d 61 70 3c 69 6e 74 2c 20 69 6e 74 3e 20 6c  .map<int, int> l
0640: 62 2c 20 75 62 3b 0d 0a 09 09 66 6f 72 20 28 69  b, ub;....for (i
0650: 6e 74 20 76 20 3d 20 30 3b 20 76 20 3c 20 4e 3b  nt v = 0; v < N;
0660: 20 2b 2b 76 29 20 7b 0d 0a 09 09 09 69 6e 74 20   ++v) {.....int 
0670: 72 20 3d 20 75 66 2e 46 69 6e 64 28 76 29 2c 20  r = uf.Find(v), 
0680: 6c 76 20 3d 20 6c 65 76 65 6c 5b 76 5d 3b 0d 0a  lv = level[v];..
0690: 09 09 09 69 66 20 28 6c 62 2e 63 6f 75 6e 74 28  ...if (lb.count(
06a0: 72 29 20 3d 3d 20 30 29 20 7b 0d 0a 09 09 09 09  r) == 0) {......
06b0: 6c 62 5b 72 5d 20 3d 20 6c 76 3b 0d 0a 09 09 09  lb[r] = lv;.....
06c0: 09 75 62 5b 72 5d 20 3d 20 6c 76 3b 0d 0a 09 09  .ub[r] = lv;....
06d0: 09 7d 0d 0a 09 09 09 65 6c 73 65 20 7b 0d 0a 09  .}.....else {...
06e0: 09 09 09 6c 62 5b 72 5d 20 3d 20 6d 69 6e 28 6c  ...lb[r] = min(l
06f0: 62 5b 72 5d 2c 20 6c 76 29 3b 0d 0a 09 09 09 09  b[r], lv);......
0700: 75 62 5b 72 5d 20 3d 20 6d 61 78 28 75 62 5b 72  ub[r] = max(ub[r
0710: 5d 2c 20 6c 76 29 3b 0d 0a 09 09 09 7d 0d 0a 09  ], lv);.....}...
0720: 09 7d 0d 0a 0d 0a 09 09 76 65 63 74 6f 72 3c 70  .}......vector<p
0730: 61 69 72 3c 69 6e 74 2c 20 69 6e 74 3e 3e 20 72  air<int, int>> r
0740: 6e 67 3b 0d 0a 09 09 66 6f 72 28 61 75 74 6f 20  ng;....for(auto 
0750: 6b 76 20 3a 20 6c 62 29 0d 0a 09 09 09 72 6e 67  kv : lb).....rng
0760: 2e 65 6d 70 6c 61 63 65 5f 62 61 63 6b 28 6b 76  .emplace_back(kv
0770: 2e 73 65 63 6f 6e 64 2c 20 75 62 5b 6b 76 2e 66  .second, ub[kv.f
0780: 69 72 73 74 5d 29 3b 0d 0a 09 09 69 6e 74 20 4c  irst]);....int L
0790: 20 3d 20 2a 6d 61 78 5f 65 6c 65 6d 65 6e 74 28   = *max_element(
07a0: 6c 65 76 65 6c 2e 62 65 67 69 6e 28 29 2c 20 6c  level.begin(), l
07b0: 65 76 65 6c 2e 65 6e 64 28 29 29 3b 0d 0a 09 09  evel.end());....
07c0: 72 65 74 75 72 6e 20 73 6f 6c 76 65 28 72 6e 67  return solve(rng
07d0: 2c 20 4c 29 3b 0d 0a 09 7d 0d 0a 0d 0a 09 2f 2f  , L);...}.....//
07e0: 20 4e 75 6d 62 65 72 20 6f 66 20 77 61 79 73 20   Number of ways 
07f0: 74 6f 20 63 6f 76 65 72 20 5b 30 2c 4c 5d 20 62  to cover [0,L] b
0800: 79 20 63 68 6f 6f 73 69 6e 67 20 73 75 62 73 65  y choosing subse
0810: 74 20 6f 66 20 7c 72 6e 67 7c 73 2e 0d 0a 09 69  t of |rng|s....i
0820: 6e 74 20 73 6f 6c 76 65 28 76 65 63 74 6f 72 3c  nt solve(vector<
0830: 70 61 69 72 3c 69 6e 74 2c 20 69 6e 74 3e 3e 20  pair<int, int>> 
0840: 72 6e 67 2c 20 69 6e 74 20 4c 29 20 7b 0d 0a 09  rng, int L) {...
0850: 09 73 6f 72 74 28 72 6e 67 2e 62 65 67 69 6e 28  .sort(rng.begin(
0860: 29 2c 20 72 6e 67 2e 65 6e 64 28 29 29 3b 0d 0a  ), rng.end());..
0870: 0d 0a 09 09 66 75 6e 63 74 69 6f 6e 3c 69 6e 74  ....function<int
0880: 28 69 6e 74 2c 20 69 6e 74 29 3e 20 72 65 63 3b  (int, int)> rec;
0890: 0d 0a 09 09 76 65 63 74 6f 72 3c 69 6e 74 3e 20  ....vector<int> 
08a0: 6d 65 6d 6f 28 72 6e 67 2e 73 69 7a 65 28 29 20  memo(rng.size() 
08b0: 2a 20 28 4c 2b 32 29 2c 20 2d 31 29 3b 0d 0a 09  * (L+2), -1);...
08c0: 09 72 65 63 20 3d 20 5b 26 5d 28 69 6e 74 20 69  .rec = [&](int i
08d0: 2c 20 69 6e 74 20 63 6f 76 65 72 65 64 55 70 74  , int coveredUpt
08e0: 6f 29 20 7b 0d 0a 09 09 09 69 66 20 28 69 20 3d  o) {.....if (i =
08f0: 3d 20 72 6e 67 2e 73 69 7a 65 28 29 29 0d 0a 09  = rng.size())...
0900: 09 09 09 72 65 74 75 72 6e 20 63 6f 76 65 72 65  ...return covere
0910: 64 55 70 74 6f 20 3d 3d 20 4c 20 3f 20 31 20 3a  dUpto == L ? 1 :
0920: 20 30 3b 0d 0a 0d 0a 09 09 09 63 6f 6e 73 74 20   0;.......const 
0930: 69 6e 74 20 6b 65 79 20 3d 20 69 2a 28 4c 20 2b  int key = i*(L +
0940: 20 31 29 20 2b 20 28 63 6f 76 65 72 65 64 55 70   1) + (coveredUp
0950: 74 6f 2b 31 29 3b 0d 0a 09 09 09 69 66 20 28 6d  to+1);.....if (m
0960: 65 6d 6f 5b 6b 65 79 5d 20 21 3d 20 2d 31 29 0d  emo[key] != -1).
0970: 0a 09 09 09 09 72 65 74 75 72 6e 20 6d 65 6d 6f  .....return memo
0980: 5b 6b 65 79 5d 3b 0d 0a 0d 0a 09 09 09 69 66 20  [key];.......if 
0990: 28 63 6f 76 65 72 65 64 55 70 74 6f 20 2b 20 32  (coveredUpto + 2
09a0: 20 3c 3d 20 72 6e 67 5b 69 5d 2e 66 69 72 73 74   <= rng[i].first
09b0: 29 0d 0a 09 09 09 09 72 65 74 75 72 6e 20 6d 65  )......return me
09c0: 6d 6f 5b 6b 65 79 5d 20 3d 20 30 3b 0d 0a 09 09  mo[key] = 0;....
09d0: 09 72 65 74 75 72 6e 20 6d 65 6d 6f 5b 6b 65 79  .return memo[key
09e0: 5d 20 3d 20 28 72 65 63 28 69 20 2b 20 31 2c 20  ] = (rec(i + 1, 
09f0: 63 6f 76 65 72 65 64 55 70 74 6f 29 20 2b 20 72  coveredUpto) + r
0a00: 65 63 28 69 20 2b 20 31 2c 20 6d 61 78 28 63 6f  ec(i + 1, max(co
0a10: 76 65 72 65 64 55 70 74 6f 2c 20 72 6e 67 5b 69  veredUpto, rng[i
0a20: 5d 2e 73 65 63 6f 6e 64 29 29 29 20 25 20 4d 4f  ].second))) % MO
0a30: 44 56 41 4c 3b 0d 0a 09 09 7d 3b 0d 0a 09 09 72  DVAL;....};....r
0a40: 65 74 75 72 6e 20 72 65 63 28 30 2c 20 2d 31 29  eturn rec(0, -1)
0a50: 3b 0d 0a 09 7d 0d 0a 7d 3b 0d 0a 0d 0a 2f 2f 20  ;...}..};....// 
0a60: 42 45 47 49 4e 20 43 55 54 20 48 45 52 45 0d 0a  BEGIN CUT HERE..
0a70: 23 69 6e 63 6c 75 64 65 20 3c 63 74 69 6d 65 3e  #include <ctime>
0a80: 0d 0a 64 6f 75 62 6c 65 20 73 74 61 72 74 5f 74  ..double start_t
0a90: 69 6d 65 3b 20 73 74 72 69 6e 67 20 74 69 6d 65  ime; string time
0aa0: 72 28 29 0d 0a 20 7b 20 6f 73 74 72 69 6e 67 73  r().. { ostrings
0ab0: 74 72 65 61 6d 20 6f 73 3b 20 6f 73 20 3c 3c 20  tream os; os << 
0ac0: 22 20 28 22 20 3c 3c 20 69 6e 74 28 28 63 6c 6f  " (" << int((clo
0ad0: 63 6b 28 29 2d 73 74 61 72 74 5f 74 69 6d 65 29  ck()-start_time)
0ae0: 2f 43 4c 4f 43 4b 53 5f 50 45 52 5f 53 45 43 2a  /CLOCKS_PER_SEC*
0af0: 31 30 30 30 29 20 3c 3c 20 22 20 6d 73 65 63 29  1000) << " msec)
0b00: 22 3b 20 72 65 74 75 72 6e 20 6f 73 2e 73 74 72  "; return os.str
0b10: 28 29 3b 20 7d 0d 0a 74 65 6d 70 6c 61 74 65 3c  (); }..template<
0b20: 74 79 70 65 6e 61 6d 65 20 54 3e 20 6f 73 74 72  typename T> ostr
0b30: 65 61 6d 26 20 6f 70 65 72 61 74 6f 72 3c 3c 28  eam& operator<<(
0b40: 6f 73 74 72 65 61 6d 26 20 6f 73 2c 20 63 6f 6e  ostream& os, con
0b50: 73 74 20 76 65 63 74 6f 72 3c 54 3e 26 20 76 29  st vector<T>& v)
0b60: 0d 0a 20 7b 20 6f 73 20 3c 3c 20 22 7b 20 22 3b  .. { os << "{ ";
0b70: 0d 0a 20 20 20 66 6f 72 28 74 79 70 65 6e 61 6d  ..   for(typenam
0b80: 65 20 76 65 63 74 6f 72 3c 54 3e 3a 3a 63 6f 6e  e vector<T>::con
0b90: 73 74 5f 69 74 65 72 61 74 6f 72 20 69 74 3d 76  st_iterator it=v
0ba0: 2e 62 65 67 69 6e 28 29 3b 20 69 74 21 3d 76 2e  .begin(); it!=v.
0bb0: 65 6e 64 28 29 3b 20 2b 2b 69 74 29 0d 0a 20 20  end(); ++it)..  
0bc0: 20 6f 73 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 2a   os << '\"' << *
0bd0: 69 74 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 28 69  it << '\"' << (i
0be0: 74 2b 31 3d 3d 76 2e 65 6e 64 28 29 20 3f 20 22  t+1==v.end() ? "
0bf0: 22 20 3a 20 22 2c 20 22 29 3b 20 6f 73 20 3c 3c  " : ", "); os <<
0c00: 20 22 20 7d 22 3b 20 72 65 74 75 72 6e 20 6f 73   " }"; return os
0c10: 3b 20 7d 0d 0a 76 6f 69 64 20 76 65 72 69 66 79  ; }..void verify
0c20: 5f 63 61 73 65 28 63 6f 6e 73 74 20 69 6e 74 26  _case(const int&
0c30: 20 45 78 70 65 63 74 65 64 2c 20 63 6f 6e 73 74   Expected, const
0c40: 20 69 6e 74 26 20 52 65 63 65 69 76 65 64 29 20   int& Received) 
0c50: 7b 0d 0a 20 62 6f 6f 6c 20 6f 6b 20 3d 20 28 45  {.. bool ok = (E
0c60: 78 70 65 63 74 65 64 20 3d 3d 20 52 65 63 65 69  xpected == Recei
0c70: 76 65 64 29 3b 0d 0a 20 69 66 28 6f 6b 29 20 63  ved);.. if(ok) c
0c80: 65 72 72 20 3c 3c 20 22 50 41 53 53 45 44 22 20  err << "PASSED" 
0c90: 3c 3c 20 74 69 6d 65 72 28 29 20 3c 3c 20 65 6e  << timer() << en
0ca0: 64 6c 3b 20 20 65 6c 73 65 20 7b 20 63 65 72 72  dl;  else { cerr
0cb0: 20 3c 3c 20 22 46 41 49 4c 45 44 22 20 3c 3c 20   << "FAILED" << 
0cc0: 74 69 6d 65 72 28 29 20 3c 3c 20 65 6e 64 6c 3b  timer() << endl;
0cd0: 0d 0a 20 63 65 72 72 20 3c 3c 20 22 5c 74 6f 3a  .. cerr << "\to:
0ce0: 20 5c 22 22 20 3c 3c 20 45 78 70 65 63 74 65 64   \"" << Expected
0cf0: 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 65 6e 64 6c   << '\"' << endl
0d00: 20 3c 3c 20 22 5c 74 78 3a 20 5c 22 22 20 3c 3c   << "\tx: \"" <<
0d10: 20 52 65 63 65 69 76 65 64 20 3c 3c 20 27 5c 22   Received << '\"
0d20: 27 20 3c 3c 20 65 6e 64 6c 3b 20 7d 20 7d 0d 0a  ' << endl; } }..
0d30: 23 64 65 66 69 6e 65 20 43 41 53 45 28 4e 29 20  #define CASE(N) 
0d40: 7b 63 65 72 72 20 3c 3c 20 22 54 65 73 74 20 43  {cerr << "Test C
0d50: 61 73 65 20 23 22 20 3c 3c 20 4e 20 3c 3c 20 22  ase #" << N << "
0d60: 2e 2e 2e 22 20 3c 3c 20 66 6c 75 73 68 3b 20 73  ..." << flush; s
0d70: 74 61 72 74 5f 74 69 6d 65 3d 63 6c 6f 63 6b 28  tart_time=clock(
0d80: 29 3b 0d 0a 23 64 65 66 69 6e 65 20 45 4e 44 09  );..#define END.
0d90: 20 76 65 72 69 66 79 5f 63 61 73 65 28 5f 2c 20   verify_case(_, 
0da0: 44 65 6e 67 6b 6c 65 6b 47 61 6e 65 73 68 41 6e  DengklekGaneshAn
0db0: 64 54 72 65 65 28 29 2e 67 65 74 43 6f 75 6e 74  dTree().getCount
0dc0: 28 6c 61 62 65 6c 73 2c 20 70 61 72 65 6e 74 73  (labels, parents
0dd0: 29 29 3b 7d 0d 0a 69 6e 74 20 6d 61 69 6e 28 29  ));}..int main()
0de0: 7b 0d 0a 0d 0a 43 41 53 45 28 30 29 0d 0a 09 73  {....CASE(0)...s
0df0: 74 72 69 6e 67 20 6c 61 62 65 6c 73 20 3d 20 22  tring labels = "
0e00: 73 65 65 6d 73 22 3b 20 0d 0a 09 69 6e 74 20 70  seems"; ...int p
0e10: 61 72 65 6e 74 73 5f 5b 5d 20 3d 20 7b 30 2c 20  arents_[] = {0, 
0e20: 31 2c 20 30 2c 20 33 7d 3b 0d 0a 09 20 20 76 65  1, 0, 3};...  ve
0e30: 63 74 6f 72 20 3c 69 6e 74 3e 20 70 61 72 65 6e  ctor <int> paren
0e40: 74 73 28 70 61 72 65 6e 74 73 5f 2c 20 70 61 72  ts(parents_, par
0e50: 65 6e 74 73 5f 2b 73 69 7a 65 6f 66 28 70 61 72  ents_+sizeof(par
0e60: 65 6e 74 73 5f 29 2f 73 69 7a 65 6f 66 28 2a 70  ents_)/sizeof(*p
0e70: 61 72 65 6e 74 73 5f 29 29 3b 20 0d 0a 09 69 6e  arents_)); ...in
0e80: 74 20 5f 20 3d 20 35 3b 20 0d 0a 45 4e 44 0d 0a  t _ = 5; ..END..
0e90: 43 41 53 45 28 31 29 0d 0a 09 73 74 72 69 6e 67  CASE(1)...string
0ea0: 20 6c 61 62 65 6c 73 20 3d 20 22 6c 69 6b 65 22   labels = "like"
0eb0: 3b 20 0d 0a 09 69 6e 74 20 70 61 72 65 6e 74 73  ; ...int parents
0ec0: 5f 5b 5d 20 3d 20 7b 30 2c 20 30 2c 20 30 7d 3b  _[] = {0, 0, 0};
0ed0: 0d 0a 09 20 20 76 65 63 74 6f 72 20 3c 69 6e 74  ...  vector <int
0ee0: 3e 20 70 61 72 65 6e 74 73 28 70 61 72 65 6e 74  > parents(parent
0ef0: 73 5f 2c 20 70 61 72 65 6e 74 73 5f 2b 73 69 7a  s_, parents_+siz
0f00: 65 6f 66 28 70 61 72 65 6e 74 73 5f 29 2f 73 69  eof(parents_)/si
0f10: 7a 65 6f 66 28 2a 70 61 72 65 6e 74 73 5f 29 29  zeof(*parents_))
0f20: 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 37 3b 20  ; ...int _ = 7; 
0f30: 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 32 29 0d 0a  ..END..CASE(2)..
0f40: 09 73 74 72 69 6e 67 20 6c 61 62 65 6c 73 20 3d  .string labels =
0f50: 20 22 61 22 3b 20 0d 0a 09 76 65 63 74 6f 72 20   "a"; ...vector 
0f60: 3c 69 6e 74 3e 20 70 61 72 65 6e 74 73 3b 20 0d  <int> parents; .
0f70: 0a 09 69 6e 74 20 5f 20 3d 20 31 3b 20 0d 0a 45  ..int _ = 1; ..E
0f80: 4e 44 0d 0a 43 41 53 45 28 33 29 0d 0a 09 73 74  ND..CASE(3)...st
0f90: 72 69 6e 67 20 6c 61 62 65 6c 73 20 3d 20 22 63  ring labels = "c
0fa0: 6f 69 6e 63 69 64 65 6e 63 65 22 3b 20 0d 0a 09  oincidence"; ...
0fb0: 69 6e 74 20 70 61 72 65 6e 74 73 5f 5b 5d 20 3d  int parents_[] =
0fc0: 20 7b 30 2c 20 31 2c 20 32 2c 20 30 2c 20 32 2c   {0, 1, 2, 0, 2,
0fd0: 20 31 2c 20 34 2c 20 37 2c 20 37 2c 20 36 7d 3b   1, 4, 7, 7, 6};
0fe0: 0d 0a 09 20 20 76 65 63 74 6f 72 20 3c 69 6e 74  ...  vector <int
0ff0: 3e 20 70 61 72 65 6e 74 73 28 70 61 72 65 6e 74  > parents(parent
1000: 73 5f 2c 20 70 61 72 65 6e 74 73 5f 2b 73 69 7a  s_, parents_+siz
1010: 65 6f 66 28 70 61 72 65 6e 74 73 5f 29 2f 73 69  eof(parents_)/si
1020: 7a 65 6f 66 28 2a 70 61 72 65 6e 74 73 5f 29 29  zeof(*parents_))
1030: 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 32 31 38  ; ...int _ = 218
1040: 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 34 29  ; ..END..CASE(4)
1050: 0d 0a 09 73 74 72 69 6e 67 20 6c 61 62 65 6c 73  ...string labels
1060: 20 3d 20 22 74 6f 70 63 6f 64 65 72 6f 70 65 6e   = "topcoderopen
1070: 72 6f 75 6e 64 74 77 6f 62 67 6f 6f 64 6c 75 63  roundtwobgoodluc
1080: 6b 68 61 76 65 66 75 6e 22 3b 20 0d 0a 09 69 6e  khavefun"; ...in
1090: 74 20 70 61 72 65 6e 74 73 5f 5b 5d 20 3d 20 7b  t parents_[] = {
10a0: 30 2c 20 31 2c 20 30 2c 20 31 2c 20 30 2c 20 31  0, 1, 0, 1, 0, 1
10b0: 2c 20 30 2c 20 31 2c 20 30 2c 20 31 2c 20 30 2c  , 0, 1, 0, 1, 0,
10c0: 20 31 2c 20 30 2c 20 31 2c 20 30 2c 20 31 2c 20   1, 0, 1, 0, 1, 
10d0: 30 2c 20 31 2c 20 30 2c 20 31 2c 20 30 2c 20 31  0, 1, 0, 1, 0, 1
10e0: 2c 20 30 2c 20 31 2c 20 30 2c 20 31 2c 20 30 2c  , 0, 1, 0, 1, 0,
10f0: 20 31 2c 20 30 2c 20 31 2c 20 30 2c 20 31 2c 20   1, 0, 1, 0, 1, 
1100: 30 2c 20 31 2c 20 30 7d 3b 0d 0a 09 20 20 76 65  0, 1, 0};...  ve
1110: 63 74 6f 72 20 3c 69 6e 74 3e 20 70 61 72 65 6e  ctor <int> paren
1120: 74 73 28 70 61 72 65 6e 74 73 5f 2c 20 70 61 72  ts(parents_, par
1130: 65 6e 74 73 5f 2b 73 69 7a 65 6f 66 28 70 61 72  ents_+sizeof(par
1140: 65 6e 74 73 5f 29 2f 73 69 7a 65 6f 66 28 2a 70  ents_)/sizeof(*p
1150: 61 72 65 6e 74 73 5f 29 29 3b 20 0d 0a 09 69 6e  arents_)); ...in
1160: 74 20 5f 20 3d 20 31 34 37 34 31 38 30 39 38 3b  t _ = 147418098;
1170: 20 0d 0a 45 4e 44 0d 0a 2f 2a 0d 0a 43 41 53 45   ..END../*..CASE
1180: 28 35 29 0d 0a 09 73 74 72 69 6e 67 20 6c 61 62  (5)...string lab
1190: 65 6c 73 20 3d 20 3b 20 0d 0a 09 69 6e 74 20 70  els = ; ...int p
11a0: 61 72 65 6e 74 73 5f 5b 5d 20 3d 20 3b 0d 0a 09  arents_[] = ;...
11b0: 20 20 76 65 63 74 6f 72 20 3c 69 6e 74 3e 20 70    vector <int> p
11c0: 61 72 65 6e 74 73 28 70 61 72 65 6e 74 73 5f 2c  arents(parents_,
11d0: 20 70 61 72 65 6e 74 73 5f 2b 73 69 7a 65 6f 66   parents_+sizeof
11e0: 28 70 61 72 65 6e 74 73 5f 29 2f 73 69 7a 65 6f  (parents_)/sizeo
11f0: 66 28 2a 70 61 72 65 6e 74 73 5f 29 29 3b 20 0d  f(*parents_)); .
1200: 0a 09 69 6e 74 20 5f 20 3d 20 3b 20 0d 0a 45 4e  ..int _ = ; ..EN
1210: 44 0d 0a 43 41 53 45 28 36 29 0d 0a 09 73 74 72  D..CASE(6)...str
1220: 69 6e 67 20 6c 61 62 65 6c 73 20 3d 20 3b 20 0d  ing labels = ; .
1230: 0a 09 69 6e 74 20 70 61 72 65 6e 74 73 5f 5b 5d  ..int parents_[]
1240: 20 3d 20 3b 0d 0a 09 20 20 76 65 63 74 6f 72 20   = ;...  vector 
1250: 3c 69 6e 74 3e 20 70 61 72 65 6e 74 73 28 70 61  <int> parents(pa
1260: 72 65 6e 74 73 5f 2c 20 70 61 72 65 6e 74 73 5f  rents_, parents_
1270: 2b 73 69 7a 65 6f 66 28 70 61 72 65 6e 74 73 5f  +sizeof(parents_
1280: 29 2f 73 69 7a 65 6f 66 28 2a 70 61 72 65 6e 74  )/sizeof(*parent
1290: 73 5f 29 29 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d  s_)); ...int _ =
12a0: 20 3b 20 0d 0a 45 4e 44 0d 0a 2a 2f 0d 0a 7d 0d   ; ..END..*/..}.
12b0: 0a 2f 2f 20 45 4e 44 20 43 55 54 20 48 45 52 45  .// END CUT HERE
12c0: 0d 0a                                            ..