Hex Artifact Content
Not logged in

Artifact dd4898e19fbf1df80da752a97eafdf2071c7715e:


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 63 73 74 72 69 6e 67 3e 0d 0a 75 73 69 6e 67  <cstring>..using
0150: 20 6e 61 6d 65 73 70 61 63 65 20 73 74 64 3b 0d   namespace std;.
0160: 0a 74 79 70 65 64 65 66 20 6c 6f 6e 67 20 6c 6f  .typedef long lo
0170: 6e 67 20 4c 4c 3b 0d 0a 74 79 70 65 64 65 66 20  ng LL;..typedef 
0180: 63 6f 6d 70 6c 65 78 3c 64 6f 75 62 6c 65 3e 20  complex<double> 
0190: 43 4d 50 3b 0d 0a 0d 0a 63 6c 61 73 73 20 42 6f  CMP;....class Bo
01a0: 75 6e 63 69 6e 67 42 61 6c 6c 73 20 7b 20 70 75  uncingBalls { pu
01b0: 62 6c 69 63 3a 0d 0a 09 64 6f 75 62 6c 65 20 65  blic:...double e
01c0: 78 70 65 63 74 65 64 42 6f 75 6e 63 65 73 28 76  xpectedBounces(v
01d0: 65 63 74 6f 72 20 3c 69 6e 74 3e 20 78 2c 20 69  ector <int> x, i
01e0: 6e 74 20 54 29 20 0d 0a 09 7b 0d 0a 09 09 73 6f  nt T) ...{....so
01f0: 72 74 28 78 2e 62 65 67 69 6e 28 29 2c 20 78 2e  rt(x.begin(), x.
0200: 65 6e 64 28 29 29 3b 0d 0a 0d 0a 09 09 64 6f 75  end());......dou
0210: 62 6c 65 20 20 74 6f 74 61 6c 20 3d 20 30 3b 0d  ble  total = 0;.
0220: 0a 0d 0a 09 09 69 6e 74 20 4e 20 3d 20 78 2e 73  .....int N = x.s
0230: 69 7a 65 28 29 3b 0d 0a 09 09 66 6f 72 28 69 6e  ize();....for(in
0240: 74 20 6d 3d 30 3b 20 6d 3c 28 31 3c 3c 4e 29 3b  t m=0; m<(1<<N);
0250: 20 2b 2b 6d 29 0d 0a 09 09 7b 0d 0a 09 09 09 76   ++m)....{.....v
0260: 65 63 74 6f 72 3c 62 6f 6f 6c 3e 20 67 6f 4c 65  ector<bool> goLe
0270: 66 74 28 4e 29 3b 0d 0a 09 09 09 66 6f 72 28 69  ft(N);.....for(i
0280: 6e 74 20 69 3d 30 3b 20 69 3c 4e 3b 20 2b 2b 69  nt i=0; i<N; ++i
0290: 29 0d 0a 09 09 09 09 67 6f 4c 65 66 74 5b 69 5d  )......goLeft[i]
02a0: 20 3d 20 28 28 31 3c 3c 69 29 26 6d 29 20 21 3d   = ((1<<i)&m) !=
02b0: 20 30 3b 0d 0a 0d 0a 09 09 09 74 6f 74 61 6c 20   0;.......total 
02c0: 2b 3d 20 6e 75 6d 28 78 2c 20 67 6f 4c 65 66 74  += num(x, goLeft
02d0: 2c 20 54 29 3b 0d 0a 09 09 7d 0d 0a 0d 0a 09 09  , T);....}......
02e0: 72 65 74 75 72 6e 20 74 6f 74 61 6c 20 2f 20 28  return total / (
02f0: 31 3c 3c 4e 29 3b 0d 0a 09 7d 0d 0a 0d 0a 09 64  1<<N);...}.....d
0300: 6f 75 62 6c 65 20 6e 75 6d 28 20 63 6f 6e 73 74  ouble num( const
0310: 20 76 65 63 74 6f 72 3c 69 6e 74 3e 26 20 78 2c   vector<int>& x,
0320: 20 63 6f 6e 73 74 20 76 65 63 74 6f 72 3c 62 6f   const vector<bo
0330: 6f 6c 3e 26 20 67 6f 4c 65 66 74 2c 20 69 6e 74  ol>& goLeft, int
0340: 20 54 20 29 0d 0a 09 7b 0d 0a 09 09 69 6e 74 20   T )...{....int 
0350: 63 6e 74 20 3d 20 30 3b 0d 0a 09 09 66 6f 72 28  cnt = 0;....for(
0360: 69 6e 74 20 69 3d 30 3b 20 69 3c 78 2e 73 69 7a  int i=0; i<x.siz
0370: 65 28 29 3b 20 2b 2b 69 29 0d 0a 09 09 09 66 6f  e(); ++i).....fo
0380: 72 28 69 6e 74 20 6a 3d 69 2b 31 3b 20 6a 3c 78  r(int j=i+1; j<x
0390: 2e 73 69 7a 65 28 29 3b 20 2b 2b 6a 29 0d 0a 09  .size(); ++j)...
03a0: 09 09 09 69 66 28 20 21 67 6f 4c 65 66 74 5b 69  ...if( !goLeft[i
03b0: 5d 20 26 26 20 67 6f 4c 65 66 74 5b 6a 5d 20 26  ] && goLeft[j] &
03c0: 26 20 78 5b 6a 5d 2d 78 5b 69 5d 3c 3d 32 2a 54  & x[j]-x[i]<=2*T
03d0: 20 29 0d 0a 09 09 09 09 09 2b 2b 63 6e 74 3b 0d   ).......++cnt;.
03e0: 0a 09 09 72 65 74 75 72 6e 20 63 6e 74 3b 0d 0a  ...return cnt;..
03f0: 09 7d 0d 0a 7d 3b 0d 0a 0d 0a 2f 2f 20 42 45 47  .}..};....// BEG
0400: 49 4e 20 43 55 54 20 48 45 52 45 0d 0a 23 69 6e  IN CUT HERE..#in
0410: 63 6c 75 64 65 20 3c 63 74 69 6d 65 3e 0d 0a 64  clude <ctime>..d
0420: 6f 75 62 6c 65 20 73 74 61 72 74 5f 74 69 6d 65  ouble start_time
0430: 3b 20 73 74 72 69 6e 67 20 74 69 6d 65 72 28 29  ; string timer()
0440: 0d 0a 20 7b 20 6f 73 74 72 69 6e 67 73 74 72 65  .. { ostringstre
0450: 61 6d 20 6f 73 3b 20 6f 73 20 3c 3c 20 22 20 28  am os; os << " (
0460: 22 20 3c 3c 20 69 6e 74 28 28 63 6c 6f 63 6b 28  " << int((clock(
0470: 29 2d 73 74 61 72 74 5f 74 69 6d 65 29 2f 43 4c  )-start_time)/CL
0480: 4f 43 4b 53 5f 50 45 52 5f 53 45 43 2a 31 30 30  OCKS_PER_SEC*100
0490: 30 29 20 3c 3c 20 22 20 6d 73 65 63 29 22 3b 20  0) << " msec)"; 
04a0: 72 65 74 75 72 6e 20 6f 73 2e 73 74 72 28 29 3b  return os.str();
04b0: 20 7d 0d 0a 74 65 6d 70 6c 61 74 65 3c 74 79 70   }..template<typ
04c0: 65 6e 61 6d 65 20 54 3e 20 6f 73 74 72 65 61 6d  ename T> ostream
04d0: 26 20 6f 70 65 72 61 74 6f 72 3c 3c 28 6f 73 74  & operator<<(ost
04e0: 72 65 61 6d 26 20 6f 73 2c 20 63 6f 6e 73 74 20  ream& os, const 
04f0: 76 65 63 74 6f 72 3c 54 3e 26 20 76 29 0d 0a 20  vector<T>& v).. 
0500: 7b 20 6f 73 20 3c 3c 20 22 7b 20 22 3b 0d 0a 20  { os << "{ ";.. 
0510: 20 20 66 6f 72 28 74 79 70 65 6e 61 6d 65 20 76    for(typename v
0520: 65 63 74 6f 72 3c 54 3e 3a 3a 63 6f 6e 73 74 5f  ector<T>::const_
0530: 69 74 65 72 61 74 6f 72 20 69 74 3d 76 2e 62 65  iterator it=v.be
0540: 67 69 6e 28 29 3b 20 69 74 21 3d 76 2e 65 6e 64  gin(); it!=v.end
0550: 28 29 3b 20 2b 2b 69 74 29 0d 0a 20 20 20 6f 73  (); ++it)..   os
0560: 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 2a 69 74 20   << '\"' << *it 
0570: 3c 3c 20 27 5c 22 27 20 3c 3c 20 28 69 74 2b 31  << '\"' << (it+1
0580: 3d 3d 76 2e 65 6e 64 28 29 20 3f 20 22 22 20 3a  ==v.end() ? "" :
0590: 20 22 2c 20 22 29 3b 20 6f 73 20 3c 3c 20 22 20   ", "); os << " 
05a0: 7d 22 3b 20 72 65 74 75 72 6e 20 6f 73 3b 20 7d  }"; return os; }
05b0: 0d 0a 76 6f 69 64 20 76 65 72 69 66 79 5f 63 61  ..void verify_ca
05c0: 73 65 28 63 6f 6e 73 74 20 64 6f 75 62 6c 65 26  se(const double&
05d0: 20 45 78 70 65 63 74 65 64 2c 20 63 6f 6e 73 74   Expected, const
05e0: 20 64 6f 75 62 6c 65 26 20 52 65 63 65 69 76 65   double& Receive
05f0: 64 29 20 7b 0d 0a 20 64 6f 75 62 6c 65 20 64 69  d) {.. double di
0600: 66 66 20 3d 20 45 78 70 65 63 74 65 64 20 2d 20  ff = Expected - 
0610: 52 65 63 65 69 76 65 64 3b 20 69 66 20 28 64 69  Received; if (di
0620: 66 66 20 3c 20 30 29 20 64 69 66 66 20 3d 20 2d  ff < 0) diff = -
0630: 64 69 66 66 3b 20 62 6f 6f 6c 20 6f 6b 20 3d 20  diff; bool ok = 
0640: 28 64 69 66 66 20 3c 20 31 65 2d 39 29 3b 0d 0a  (diff < 1e-9);..
0650: 20 69 66 28 6f 6b 29 20 63 65 72 72 20 3c 3c 20   if(ok) cerr << 
0660: 22 50 41 53 53 45 44 22 20 3c 3c 20 74 69 6d 65  "PASSED" << time
0670: 72 28 29 20 3c 3c 20 65 6e 64 6c 3b 20 20 65 6c  r() << endl;  el
0680: 73 65 20 7b 20 63 65 72 72 20 3c 3c 20 22 46 41  se { cerr << "FA
0690: 49 4c 45 44 22 20 3c 3c 20 74 69 6d 65 72 28 29  ILED" << timer()
06a0: 20 3c 3c 20 65 6e 64 6c 3b 0d 0a 20 63 65 72 72   << endl;.. cerr
06b0: 20 3c 3c 20 22 5c 74 6f 3a 20 5c 22 22 20 3c 3c   << "\to: \"" <<
06c0: 20 45 78 70 65 63 74 65 64 20 3c 3c 20 27 5c 22   Expected << '\"
06d0: 27 20 3c 3c 20 65 6e 64 6c 20 3c 3c 20 22 5c 74  ' << endl << "\t
06e0: 78 3a 20 5c 22 22 20 3c 3c 20 52 65 63 65 69 76  x: \"" << Receiv
06f0: 65 64 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 65 6e  ed << '\"' << en
0700: 64 6c 3b 20 7d 20 7d 0d 0a 23 64 65 66 69 6e 65  dl; } }..#define
0710: 20 43 41 53 45 28 4e 29 20 7b 63 65 72 72 20 3c   CASE(N) {cerr <
0720: 3c 20 22 54 65 73 74 20 43 61 73 65 20 23 22 20  < "Test Case #" 
0730: 3c 3c 20 4e 20 3c 3c 20 22 2e 2e 2e 22 20 3c 3c  << N << "..." <<
0740: 20 66 6c 75 73 68 3b 20 73 74 61 72 74 5f 74 69   flush; start_ti
0750: 6d 65 3d 63 6c 6f 63 6b 28 29 3b 0d 0a 23 64 65  me=clock();..#de
0760: 66 69 6e 65 20 45 4e 44 09 20 76 65 72 69 66 79  fine END. verify
0770: 5f 63 61 73 65 28 5f 2c 20 42 6f 75 6e 63 69 6e  _case(_, Bouncin
0780: 67 42 61 6c 6c 73 28 29 2e 65 78 70 65 63 74 65  gBalls().expecte
0790: 64 42 6f 75 6e 63 65 73 28 78 2c 20 54 29 29 3b  dBounces(x, T));
07a0: 7d 0d 0a 69 6e 74 20 6d 61 69 6e 28 29 7b 0d 0a  }..int main(){..
07b0: 0d 0a 43 41 53 45 28 30 29 0d 0a 09 69 6e 74 20  ..CASE(0)...int 
07c0: 78 5f 5b 5d 20 3d 20 7b 35 2c 20 38 7d 3b 0d 0a  x_[] = {5, 8};..
07d0: 09 20 20 76 65 63 74 6f 72 20 3c 69 6e 74 3e 20  .  vector <int> 
07e0: 78 28 78 5f 2c 20 78 5f 2b 73 69 7a 65 6f 66 28  x(x_, x_+sizeof(
07f0: 78 5f 29 2f 73 69 7a 65 6f 66 28 2a 78 5f 29 29  x_)/sizeof(*x_))
0800: 3b 20 0d 0a 09 69 6e 74 20 54 20 3d 20 32 3b 20  ; ...int T = 2; 
0810: 0d 0a 09 64 6f 75 62 6c 65 20 5f 20 3d 20 30 2e  ...double _ = 0.
0820: 32 35 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 28  25; ..END..CASE(
0830: 31 29 0d 0a 09 69 6e 74 20 78 5f 5b 5d 20 3d 20  1)...int x_[] = 
0840: 7b 35 2c 20 38 7d 3b 0d 0a 09 20 20 76 65 63 74  {5, 8};...  vect
0850: 6f 72 20 3c 69 6e 74 3e 20 78 28 78 5f 2c 20 78  or <int> x(x_, x
0860: 5f 2b 73 69 7a 65 6f 66 28 78 5f 29 2f 73 69 7a  _+sizeof(x_)/siz
0870: 65 6f 66 28 2a 78 5f 29 29 3b 20 0d 0a 09 69 6e  eof(*x_)); ...in
0880: 74 20 54 20 3d 20 31 3b 20 0d 0a 09 64 6f 75 62  t T = 1; ...doub
0890: 6c 65 20 5f 20 3d 20 30 2e 30 3b 20 0d 0a 45 4e  le _ = 0.0; ..EN
08a0: 44 0d 0a 43 41 53 45 28 32 29 0d 0a 09 69 6e 74  D..CASE(2)...int
08b0: 20 78 5f 5b 5d 20 3d 20 7b 39 31 2c 20 38 35 37   x_[] = {91, 857
08c0: 2c 20 36 39 32 2c 20 35 34 2c 20 38 36 37 39 2c  , 692, 54, 8679,
08d0: 20 38 33 2c 20 37 39 32 2c 20 38 36 2c 20 39 35   83, 792, 86, 95
08e0: 33 37 2c 20 39 31 33 2c 20 36 34 2c 20 35 39 32  37, 913, 64, 592
08f0: 7d 3b 0d 0a 09 20 20 76 65 63 74 6f 72 20 3c 69  };...  vector <i
0900: 6e 74 3e 20 78 28 78 5f 2c 20 78 5f 2b 73 69 7a  nt> x(x_, x_+siz
0910: 65 6f 66 28 78 5f 29 2f 73 69 7a 65 6f 66 28 2a  eof(x_)/sizeof(*
0920: 78 5f 29 29 3b 20 0d 0a 09 69 6e 74 20 54 20 3d  x_)); ...int T =
0930: 20 34 35 38 3b 20 0d 0a 09 64 6f 75 62 6c 65 20   458; ...double 
0940: 5f 20 3d 20 31 31 2e 35 3b 20 0d 0a 45 4e 44 0d  _ = 11.5; ..END.
0950: 0a 43 41 53 45 28 33 29 0d 0a 09 69 6e 74 20 78  .CASE(3)...int x
0960: 5f 5b 5d 20 3d 20 7b 37 35 34 33 32 7d 3b 0d 0a  _[] = {75432};..
0970: 09 20 20 76 65 63 74 6f 72 20 3c 69 6e 74 3e 20  .  vector <int> 
0980: 78 28 78 5f 2c 20 78 5f 2b 73 69 7a 65 6f 66 28  x(x_, x_+sizeof(
0990: 78 5f 29 2f 73 69 7a 65 6f 66 28 2a 78 5f 29 29  x_)/sizeof(*x_))
09a0: 3b 20 0d 0a 09 69 6e 74 20 54 20 3d 20 33 38 36  ; ...int T = 386
09b0: 3b 20 0d 0a 09 64 6f 75 62 6c 65 20 5f 20 3d 20  ; ...double _ = 
09c0: 30 2e 30 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 45  0.0; ..END..CASE
09d0: 28 34 29 0d 0a 09 69 6e 74 20 78 5f 5b 5d 20 3d  (4)...int x_[] =
09e0: 20 7b 31 2c 20 32 2c 20 33 2c 20 34 2c 20 35 2c   {1, 2, 3, 4, 5,
09f0: 20 36 2c 20 37 2c 20 38 2c 20 39 2c 20 31 30 2c   6, 7, 8, 9, 10,
0a00: 20 31 31 2c 20 31 32 7d 3b 0d 0a 09 20 20 76 65   11, 12};...  ve
0a10: 63 74 6f 72 20 3c 69 6e 74 3e 20 78 28 78 5f 2c  ctor <int> x(x_,
0a20: 20 78 5f 2b 73 69 7a 65 6f 66 28 78 5f 29 2f 73   x_+sizeof(x_)/s
0a30: 69 7a 65 6f 66 28 2a 78 5f 29 29 3b 20 0d 0a 09  izeof(*x_)); ...
0a40: 69 6e 74 20 54 20 3d 20 33 3b 20 0d 0a 09 64 6f  int T = 3; ...do
0a50: 75 62 6c 65 20 5f 20 3d 20 31 32 2e 37 35 3b 20  uble _ = 12.75; 
0a60: 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 35 29 0d 0a  ..END..CASE(5)..
0a70: 09 69 6e 74 20 78 5f 5b 5d 20 3d 20 7b 31 7d 3b  .int x_[] = {1};
0a80: 0d 0a 09 20 20 76 65 63 74 6f 72 20 3c 69 6e 74  ...  vector <int
0a90: 3e 20 78 28 78 5f 2c 20 78 5f 2b 73 69 7a 65 6f  > x(x_, x_+sizeo
0aa0: 66 28 78 5f 29 2f 73 69 7a 65 6f 66 28 2a 78 5f  f(x_)/sizeof(*x_
0ab0: 29 29 3b 20 0d 0a 09 69 6e 74 20 54 20 3d 20 31  )); ...int T = 1
0ac0: 30 30 30 30 30 30 30 30 3b 20 0d 0a 09 64 6f 75  00000000; ...dou
0ad0: 62 6c 65 20 5f 20 3d 20 30 3b 20 0d 0a 45 4e 44  ble _ = 0; ..END
0ae0: 0d 0a 43 41 53 45 28 36 29 0d 0a 09 69 6e 74 20  ..CASE(6)...int 
0af0: 78 5f 5b 5d 20 3d 20 7b 31 30 30 30 30 30 30 30  x_[] = {10000000
0b00: 7d 3b 0d 0a 09 20 20 76 65 63 74 6f 72 20 3c 69  };...  vector <i
0b10: 6e 74 3e 20 78 28 78 5f 2c 20 78 5f 2b 73 69 7a  nt> x(x_, x_+siz
0b20: 65 6f 66 28 78 5f 29 2f 73 69 7a 65 6f 66 28 2a  eof(x_)/sizeof(*
0b30: 78 5f 29 29 3b 20 0d 0a 09 69 6e 74 20 54 20 3d  x_)); ...int T =
0b40: 20 31 30 30 30 30 30 3b 20 0d 0a 09 64 6f 75 62   100000; ...doub
0b50: 6c 65 20 5f 20 3d 20 30 3b 20 0d 0a 45 4e 44 0d  le _ = 0; ..END.
0b60: 0a 0d 0a 7d 0d 0a 2f 2f 20 45 4e 44 20 43 55 54  ...}..// END CUT
0b70: 20 48 45 52 45 0d 0a                              HERE..