Hex Artifact Content
Not logged in

Artifact 1b6c222f9c3582e7bf8556f33429d8f0c96d575f:


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 43 6f  CMP;....class Co
01a0: 6c 6f 72 66 75 6c 52 61 62 62 69 74 73 20 7b 20  lorfulRabbits { 
01b0: 70 75 62 6c 69 63 3a 0d 0a 09 69 6e 74 20 67 65  public:...int ge
01c0: 74 4d 69 6e 69 6d 75 6d 28 76 65 63 74 6f 72 20  tMinimum(vector 
01d0: 3c 69 6e 74 3e 20 72 65 70 6c 69 65 73 29 20 0d  <int> replies) .
01e0: 0a 09 7b 0d 0a 09 09 6d 61 70 3c 69 6e 74 2c 69  ..{....map<int,i
01f0: 6e 74 3e 20 6d 3b 0d 0a 09 09 66 6f 72 28 69 6e  nt> m;....for(in
0200: 74 20 69 3d 30 3b 20 69 3c 72 65 70 6c 69 65 73  t i=0; i<replies
0210: 2e 73 69 7a 65 28 29 3b 20 2b 2b 69 29 0d 0a 09  .size(); ++i)...
0220: 09 09 6d 5b 72 65 70 6c 69 65 73 5b 69 5d 5d 2b  ..m[replies[i]]+
0230: 2b 3b 0d 0a 09 09 69 6e 74 20 74 6f 74 61 6c 20  +;....int total 
0240: 3d 20 30 3b 0d 0a 09 09 66 6f 72 28 6d 61 70 3c  = 0;....for(map<
0250: 69 6e 74 2c 69 6e 74 3e 3a 3a 69 74 65 72 61 74  int,int>::iterat
0260: 6f 72 20 69 74 3d 6d 2e 62 65 67 69 6e 28 29 3b  or it=m.begin();
0270: 20 69 74 21 3d 6d 2e 65 6e 64 28 29 3b 20 2b 2b   it!=m.end(); ++
0280: 69 74 29 0d 0a 09 09 7b 0d 0a 09 09 09 69 6e 74  it)....{.....int
0290: 20 73 20 3d 20 69 74 2d 3e 66 69 72 73 74 20 2b   s = it->first +
02a0: 20 31 3b 0d 0a 09 09 09 69 6e 74 20 6e 20 3d 20   1;.....int n = 
02b0: 69 74 2d 3e 73 65 63 6f 6e 64 3b 0d 0a 09 09 09  it->second;.....
02c0: 74 6f 74 61 6c 20 2b 3d 20 28 6e 2b 73 2d 31 29  total += (n+s-1)
02d0: 20 2f 20 73 20 2a 20 73 3b 0d 0a 09 09 7d 0d 0a   / s * s;....}..
02e0: 09 09 72 65 74 75 72 6e 20 74 6f 74 61 6c 3b 0d  ..return total;.
02f0: 0a 09 7d 0d 0a 7d 3b 0d 0a 0d 0a 2f 2f 20 42 45  ..}..};....// BE
0300: 47 49 4e 20 43 55 54 20 48 45 52 45 0d 0a 23 69  GIN CUT HERE..#i
0310: 6e 63 6c 75 64 65 20 3c 63 74 69 6d 65 3e 0d 0a  nclude <ctime>..
0320: 64 6f 75 62 6c 65 20 73 74 61 72 74 5f 74 69 6d  double start_tim
0330: 65 3b 20 73 74 72 69 6e 67 20 74 69 6d 65 72 28  e; string timer(
0340: 29 0d 0a 20 7b 20 6f 73 74 72 69 6e 67 73 74 72  ).. { ostringstr
0350: 65 61 6d 20 6f 73 3b 20 6f 73 20 3c 3c 20 22 20  eam os; os << " 
0360: 28 22 20 3c 3c 20 69 6e 74 28 28 63 6c 6f 63 6b  (" << int((clock
0370: 28 29 2d 73 74 61 72 74 5f 74 69 6d 65 29 2f 43  ()-start_time)/C
0380: 4c 4f 43 4b 53 5f 50 45 52 5f 53 45 43 2a 31 30  LOCKS_PER_SEC*10
0390: 30 30 29 20 3c 3c 20 22 20 6d 73 65 63 29 22 3b  00) << " msec)";
03a0: 20 72 65 74 75 72 6e 20 6f 73 2e 73 74 72 28 29   return os.str()
03b0: 3b 20 7d 0d 0a 74 65 6d 70 6c 61 74 65 3c 74 79  ; }..template<ty
03c0: 70 65 6e 61 6d 65 20 54 3e 20 6f 73 74 72 65 61  pename T> ostrea
03d0: 6d 26 20 6f 70 65 72 61 74 6f 72 3c 3c 28 6f 73  m& operator<<(os
03e0: 74 72 65 61 6d 26 20 6f 73 2c 20 63 6f 6e 73 74  tream& os, const
03f0: 20 76 65 63 74 6f 72 3c 54 3e 26 20 76 29 0d 0a   vector<T>& v)..
0400: 20 7b 20 6f 73 20 3c 3c 20 22 7b 20 22 3b 0d 0a   { os << "{ ";..
0410: 20 20 20 66 6f 72 28 74 79 70 65 6e 61 6d 65 20     for(typename 
0420: 76 65 63 74 6f 72 3c 54 3e 3a 3a 63 6f 6e 73 74  vector<T>::const
0430: 5f 69 74 65 72 61 74 6f 72 20 69 74 3d 76 2e 62  _iterator it=v.b
0440: 65 67 69 6e 28 29 3b 20 69 74 21 3d 76 2e 65 6e  egin(); it!=v.en
0450: 64 28 29 3b 20 2b 2b 69 74 29 0d 0a 20 20 20 6f  d(); ++it)..   o
0460: 73 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 2a 69 74  s << '\"' << *it
0470: 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 28 69 74 2b   << '\"' << (it+
0480: 31 3d 3d 76 2e 65 6e 64 28 29 20 3f 20 22 22 20  1==v.end() ? "" 
0490: 3a 20 22 2c 20 22 29 3b 20 6f 73 20 3c 3c 20 22  : ", "); os << "
04a0: 20 7d 22 3b 20 72 65 74 75 72 6e 20 6f 73 3b 20   }"; return os; 
04b0: 7d 0d 0a 76 6f 69 64 20 76 65 72 69 66 79 5f 63  }..void verify_c
04c0: 61 73 65 28 63 6f 6e 73 74 20 69 6e 74 26 20 45  ase(const int& E
04d0: 78 70 65 63 74 65 64 2c 20 63 6f 6e 73 74 20 69  xpected, const i
04e0: 6e 74 26 20 52 65 63 65 69 76 65 64 29 20 7b 0d  nt& Received) {.
04f0: 0a 20 62 6f 6f 6c 20 6f 6b 20 3d 20 28 45 78 70  . bool ok = (Exp
0500: 65 63 74 65 64 20 3d 3d 20 52 65 63 65 69 76 65  ected == Receive
0510: 64 29 3b 0d 0a 20 69 66 28 6f 6b 29 20 63 65 72  d);.. if(ok) cer
0520: 72 20 3c 3c 20 22 50 41 53 53 45 44 22 20 3c 3c  r << "PASSED" <<
0530: 20 74 69 6d 65 72 28 29 20 3c 3c 20 65 6e 64 6c   timer() << endl
0540: 3b 20 20 65 6c 73 65 20 7b 20 63 65 72 72 20 3c  ;  else { cerr <
0550: 3c 20 22 46 41 49 4c 45 44 22 20 3c 3c 20 74 69  < "FAILED" << ti
0560: 6d 65 72 28 29 20 3c 3c 20 65 6e 64 6c 3b 0d 0a  mer() << endl;..
0570: 20 63 65 72 72 20 3c 3c 20 22 5c 74 6f 3a 20 5c   cerr << "\to: \
0580: 22 22 20 3c 3c 20 45 78 70 65 63 74 65 64 20 3c  "" << Expected <
0590: 3c 20 27 5c 22 27 20 3c 3c 20 65 6e 64 6c 20 3c  < '\"' << endl <
05a0: 3c 20 22 5c 74 78 3a 20 5c 22 22 20 3c 3c 20 52  < "\tx: \"" << R
05b0: 65 63 65 69 76 65 64 20 3c 3c 20 27 5c 22 27 20  eceived << '\"' 
05c0: 3c 3c 20 65 6e 64 6c 3b 20 7d 20 7d 0d 0a 23 64  << endl; } }..#d
05d0: 65 66 69 6e 65 20 43 41 53 45 28 4e 29 20 7b 63  efine CASE(N) {c
05e0: 65 72 72 20 3c 3c 20 22 54 65 73 74 20 43 61 73  err << "Test Cas
05f0: 65 20 23 22 20 3c 3c 20 4e 20 3c 3c 20 22 2e 2e  e #" << N << "..
0600: 2e 22 20 3c 3c 20 66 6c 75 73 68 3b 20 73 74 61  ." << flush; sta
0610: 72 74 5f 74 69 6d 65 3d 63 6c 6f 63 6b 28 29 3b  rt_time=clock();
0620: 0d 0a 23 64 65 66 69 6e 65 20 45 4e 44 09 20 76  ..#define END. v
0630: 65 72 69 66 79 5f 63 61 73 65 28 5f 2c 20 43 6f  erify_case(_, Co
0640: 6c 6f 72 66 75 6c 52 61 62 62 69 74 73 28 29 2e  lorfulRabbits().
0650: 67 65 74 4d 69 6e 69 6d 75 6d 28 72 65 70 6c 69  getMinimum(repli
0660: 65 73 29 29 3b 7d 0d 0a 69 6e 74 20 6d 61 69 6e  es));}..int main
0670: 28 29 7b 0d 0a 0d 0a 43 41 53 45 28 30 29 0d 0a  (){....CASE(0)..
0680: 09 69 6e 74 20 72 65 70 6c 69 65 73 5f 5b 5d 20  .int replies_[] 
0690: 3d 20 7b 20 31 2c 20 31 2c 20 32 2c 20 32 20 7d  = { 1, 1, 2, 2 }
06a0: 0d 0a 3b 0d 0a 09 20 20 76 65 63 74 6f 72 20 3c  ..;...  vector <
06b0: 69 6e 74 3e 20 72 65 70 6c 69 65 73 28 72 65 70  int> replies(rep
06c0: 6c 69 65 73 5f 2c 20 72 65 70 6c 69 65 73 5f 2b  lies_, replies_+
06d0: 73 69 7a 65 6f 66 28 72 65 70 6c 69 65 73 5f 29  sizeof(replies_)
06e0: 2f 73 69 7a 65 6f 66 28 2a 72 65 70 6c 69 65 73  /sizeof(*replies
06f0: 5f 29 29 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20  _)); ...int _ = 
0700: 35 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 31  5; ..END..CASE(1
0710: 29 0d 0a 09 69 6e 74 20 72 65 70 6c 69 65 73 5f  )...int replies_
0720: 5b 5d 20 3d 20 7b 20 30 20 7d 0d 0a 3b 0d 0a 09  [] = { 0 }..;...
0730: 20 20 76 65 63 74 6f 72 20 3c 69 6e 74 3e 20 72    vector <int> r
0740: 65 70 6c 69 65 73 28 72 65 70 6c 69 65 73 5f 2c  eplies(replies_,
0750: 20 72 65 70 6c 69 65 73 5f 2b 73 69 7a 65 6f 66   replies_+sizeof
0760: 28 72 65 70 6c 69 65 73 5f 29 2f 73 69 7a 65 6f  (replies_)/sizeo
0770: 66 28 2a 72 65 70 6c 69 65 73 5f 29 29 3b 20 0d  f(*replies_)); .
0780: 0a 09 69 6e 74 20 5f 20 3d 20 31 3b 20 0d 0a 45  ..int _ = 1; ..E
0790: 4e 44 0d 0a 43 41 53 45 28 32 29 0d 0a 09 69 6e  ND..CASE(2)...in
07a0: 74 20 72 65 70 6c 69 65 73 5f 5b 5d 20 3d 20 7b  t replies_[] = {
07b0: 20 32 2c 20 32 2c 20 34 34 2c 20 32 2c 20 32 2c   2, 2, 44, 2, 2,
07c0: 20 32 2c 20 34 34 34 2c 20 32 2c 20 32 20 7d 0d   2, 444, 2, 2 }.
07d0: 0a 3b 0d 0a 09 20 20 76 65 63 74 6f 72 20 3c 69  .;...  vector <i
07e0: 6e 74 3e 20 72 65 70 6c 69 65 73 28 72 65 70 6c  nt> replies(repl
07f0: 69 65 73 5f 2c 20 72 65 70 6c 69 65 73 5f 2b 73  ies_, replies_+s
0800: 69 7a 65 6f 66 28 72 65 70 6c 69 65 73 5f 29 2f  izeof(replies_)/
0810: 73 69 7a 65 6f 66 28 2a 72 65 70 6c 69 65 73 5f  sizeof(*replies_
0820: 29 29 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 34  )); ...int _ = 4
0830: 39 39 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 28  99; ..END..CASE(
0840: 33 29 0d 0a 09 69 6e 74 20 72 65 70 6c 69 65 73  3)...int replies
0850: 5f 5b 5d 20 3d 20 7b 31 7d 3b 0d 0a 09 20 20 76  _[] = {1};...  v
0860: 65 63 74 6f 72 20 3c 69 6e 74 3e 20 72 65 70 6c  ector <int> repl
0870: 69 65 73 28 72 65 70 6c 69 65 73 5f 2c 20 72 65  ies(replies_, re
0880: 70 6c 69 65 73 5f 2b 73 69 7a 65 6f 66 28 72 65  plies_+sizeof(re
0890: 70 6c 69 65 73 5f 29 2f 73 69 7a 65 6f 66 28 2a  plies_)/sizeof(*
08a0: 72 65 70 6c 69 65 73 5f 29 29 3b 20 0d 0a 09 69  replies_)); ...i
08b0: 6e 74 20 5f 20 3d 20 32 3b 20 0d 0a 45 4e 44 0d  nt _ = 2; ..END.
08c0: 0a 43 41 53 45 28 34 29 0d 0a 09 69 6e 74 20 72  .CASE(4)...int r
08d0: 65 70 6c 69 65 73 5f 5b 5d 20 3d 20 7b 32 7d 3b  eplies_[] = {2};
08e0: 0d 0a 09 20 20 76 65 63 74 6f 72 20 3c 69 6e 74  ...  vector <int
08f0: 3e 20 72 65 70 6c 69 65 73 28 72 65 70 6c 69 65  > replies(replie
0900: 73 5f 2c 20 72 65 70 6c 69 65 73 5f 2b 73 69 7a  s_, replies_+siz
0910: 65 6f 66 28 72 65 70 6c 69 65 73 5f 29 2f 73 69  eof(replies_)/si
0920: 7a 65 6f 66 28 2a 72 65 70 6c 69 65 73 5f 29 29  zeof(*replies_))
0930: 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 33 3b 20  ; ...int _ = 3; 
0940: 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 35 29 0d 0a  ..END..CASE(5)..
0950: 09 69 6e 74 20 72 65 70 6c 69 65 73 5f 5b 5d 20  .int replies_[] 
0960: 3d 20 7b 31 30 30 30 30 30 30 2c 31 30 30 30 30  = {1000000,10000
0970: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0980: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0990: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
09a0: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
09b0: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
09c0: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
09d0: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
09e0: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
09f0: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0a00: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0a10: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0a20: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0a30: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0a40: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0a50: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0a60: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0a70: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0a80: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0a90: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0aa0: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0ab0: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0ac0: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0ad0: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0ae0: 30 30 2c 31 30 30 30 30 30 30 2c 31 30 30 30 30  00,1000000,10000
0af0: 30 30 7d 3b 0d 0a 09 20 20 76 65 63 74 6f 72 20  00};...  vector 
0b00: 3c 69 6e 74 3e 20 72 65 70 6c 69 65 73 28 72 65  <int> replies(re
0b10: 70 6c 69 65 73 5f 2c 20 72 65 70 6c 69 65 73 5f  plies_, replies_
0b20: 2b 73 69 7a 65 6f 66 28 72 65 70 6c 69 65 73 5f  +sizeof(replies_
0b30: 29 2f 73 69 7a 65 6f 66 28 2a 72 65 70 6c 69 65  )/sizeof(*replie
0b40: 73 5f 29 29 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d  s_)); ...int _ =
0b50: 20 2d 31 3b 20 0d 0a 45 4e 44 0d 0a 0d 0a 7d 0d   -1; ..END....}.
0b60: 0a 2f 2f 20 45 4e 44 20 43 55 54 20 48 45 52 45  .// END CUT HERE
0b70: 0d 0a                                            ..