Artifact 75343b8007c95b1302ab93cd6c1ab13196afba86:
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 63 6f 6d 70 ..#include <comp
00d0: 6c 65 78 3e 0d 0a 23 69 6e 63 6c 75 64 65 20 3c lex>..#include <
00e0: 71 75 65 75 65 3e 0d 0a 23 69 6e 63 6c 75 64 65 queue>..#include
00f0: 20 3c 73 74 61 63 6b 3e 0d 0a 23 69 6e 63 6c 75 <stack>..#inclu
0100: 64 65 20 3c 63 6d 61 74 68 3e 0d 0a 75 73 69 6e de <cmath>..usin
0110: 67 20 6e 61 6d 65 73 70 61 63 65 20 73 74 64 3b g namespace std;
0120: 0d 0a 74 79 70 65 64 65 66 20 6c 6f 6e 67 20 6c ..typedef long l
0130: 6f 6e 67 20 4c 4c 3b 0d 0a 0d 0a 63 6c 61 73 73 ong LL;....class
0140: 20 53 74 6f 6e 65 47 61 6d 65 0d 0a 7b 0d 0a 70 StoneGame..{..p
0150: 75 62 6c 69 63 3a 0d 0a 09 69 6e 74 20 67 65 74 ublic:...int get
0160: 53 63 6f 72 65 28 76 65 63 74 6f 72 20 3c 69 6e Score(vector <in
0170: 74 3e 20 74 72 65 61 73 75 72 65 2c 20 76 65 63 t> treasure, vec
0180: 74 6f 72 20 3c 69 6e 74 3e 20 73 74 6f 6e 65 73 tor <int> stones
0190: 29 20 0d 0a 09 7b 0d 0a 09 09 2f 2f 20 4e 6f 74 ) ...{....// Not
01a0: 65 20 74 68 61 74 20 77 65 20 61 6c 77 61 79 73 e that we always
01b0: 20 6c 6f 73 65 20 27 65 76 65 6e 27 20 70 69 6c lose 'even' pil
01c0: 65 73 2e 2e 2e 0d 0a 0d 0a 09 09 76 65 63 74 6f es.........vecto
01d0: 72 3c 69 6e 74 3e 20 6f 6e 65 73 3b 0d 0a 09 09 r<int> ones;....
01e0: 69 6e 74 20 6e 75 6d 4f 64 64 73 20 3d 20 30 3b int numOdds = 0;
01f0: 0d 0a 09 09 69 6e 74 20 73 75 6d 47 74 31 20 20 ....int sumGt1
0200: 3d 20 30 3b 0d 0a 0d 0a 09 09 69 6e 74 20 6e 20 = 0;......int n
0210: 3d 20 74 72 65 61 73 75 72 65 2e 73 69 7a 65 28 = treasure.size(
0220: 29 3b 0d 0a 09 09 66 6f 72 28 69 6e 74 20 69 3d );....for(int i=
0230: 30 3b 20 69 3c 6e 3b 20 2b 2b 69 29 0d 0a 09 09 0; i<n; ++i)....
0240: 09 69 66 28 20 73 74 6f 6e 65 73 5b 69 5d 20 3d .if( stones[i] =
0250: 3d 20 31 20 29 0d 0a 09 09 09 09 6f 6e 65 73 2e = 1 )......ones.
0260: 70 75 73 68 5f 62 61 63 6b 28 20 74 72 65 61 73 push_back( treas
0270: 75 72 65 5b 69 5d 20 29 3b 0d 0a 09 09 09 65 6c ure[i] );.....el
0280: 73 65 0d 0a 09 09 09 09 6e 75 6d 4f 64 64 73 20 se......numOdds
0290: 2b 3d 20 73 74 6f 6e 65 73 5b 69 5d 25 32 2c 0d += stones[i]%2,.
02a0: 0a 09 09 09 09 73 75 6d 47 74 31 20 20 2b 3d 20 .....sumGt1 +=
02b0: 74 72 65 61 73 75 72 65 5b 69 5d 3b 0d 0a 0d 0a treasure[i];....
02c0: 09 09 73 6f 72 74 28 6f 6e 65 73 2e 72 62 65 67 ..sort(ones.rbeg
02d0: 69 6e 28 29 2c 20 6f 6e 65 73 2e 72 65 6e 64 28 in(), ones.rend(
02e0: 29 29 3b 0d 0a 0d 0a 09 09 69 6e 74 20 73 63 6f ));......int sco
02f0: 72 65 20 3d 20 30 3b 0d 0a 09 09 66 6f 72 28 69 re = 0;....for(i
0300: 6e 74 20 69 3d 30 3b 20 69 3c 6f 6e 65 73 2e 73 nt i=0; i<ones.s
0310: 69 7a 65 28 29 3b 20 69 2b 3d 32 29 20 2f 2f 20 ize(); i+=2) //
0320: 74 61 6b 65 20 31 2d 73 74 6f 6e 65 20 70 69 6c take 1-stone pil
0330: 65 73 20 61 6c 74 65 6e 61 74 69 76 65 6c 79 0d es altenatively.
0340: 0a 09 09 09 73 63 6f 72 65 20 2b 3d 20 6f 6e 65 ....score += one
0350: 73 5b 69 5d 3b 0d 0a 09 09 69 66 28 20 20 28 6f s[i];....if( (o
0360: 6e 65 73 2e 73 69 7a 65 28 29 2b 6e 75 6d 4f 64 nes.size()+numOd
0370: 64 73 29 25 32 20 3d 3d 20 31 20 29 20 2f 2f 20 ds)%2 == 1 ) //
0380: 74 68 65 6e 20 6d 61 6b 65 20 6f 64 64 2d 73 74 then make odd-st
0390: 6f 6e 65 20 70 69 6c 65 73 20 69 6e 74 6f 20 65 one piles into e
03a0: 76 65 6e 0d 0a 09 09 09 73 63 6f 72 65 20 2b 3d ven.....score +=
03b0: 20 73 75 6d 47 74 31 3b 20 2f 2f 20 69 66 20 61 sumGt1; // if a
03c0: 6c 6c 20 70 69 6c 65 73 20 61 72 65 20 6d 61 64 ll piles are mad
03d0: 65 20 65 76 65 6e 2c 20 6f 6e 65 2d 73 69 64 65 e even, one-side
03e0: 20 74 61 6b 65 73 20 61 6c 6c 0d 0a 09 09 72 65 takes all....re
03f0: 74 75 72 6e 20 73 63 6f 72 65 3b 0d 0a 09 7d 0d turn score;...}.
0400: 0a 0d 0a 2f 2f 20 42 45 47 49 4e 20 43 55 54 20 ...// BEGIN CUT
0410: 48 45 52 45 0d 0a 09 70 75 62 6c 69 63 3a 0d 0a HERE...public:..
0420: 09 76 6f 69 64 20 72 75 6e 5f 74 65 73 74 28 69 .void run_test(i
0430: 6e 74 20 43 61 73 65 29 20 7b 20 69 66 20 28 28 nt Case) { if ((
0440: 43 61 73 65 20 3d 3d 20 2d 31 29 20 7c 7c 20 28 Case == -1) || (
0450: 43 61 73 65 20 3d 3d 20 30 29 29 20 74 65 73 74 Case == 0)) test
0460: 5f 63 61 73 65 5f 30 28 29 3b 20 69 66 20 28 28 _case_0(); if ((
0470: 43 61 73 65 20 3d 3d 20 2d 31 29 20 7c 7c 20 28 Case == -1) || (
0480: 43 61 73 65 20 3d 3d 20 31 29 29 20 74 65 73 74 Case == 1)) test
0490: 5f 63 61 73 65 5f 31 28 29 3b 20 69 66 20 28 28 _case_1(); if ((
04a0: 43 61 73 65 20 3d 3d 20 2d 31 29 20 7c 7c 20 28 Case == -1) || (
04b0: 43 61 73 65 20 3d 3d 20 32 29 29 20 74 65 73 74 Case == 2)) test
04c0: 5f 63 61 73 65 5f 32 28 29 3b 20 69 66 20 28 28 _case_2(); if ((
04d0: 43 61 73 65 20 3d 3d 20 2d 31 29 20 7c 7c 20 28 Case == -1) || (
04e0: 43 61 73 65 20 3d 3d 20 33 29 29 20 74 65 73 74 Case == 3)) test
04f0: 5f 63 61 73 65 5f 33 28 29 3b 20 7d 0d 0a 09 70 _case_3(); }...p
0500: 72 69 76 61 74 65 3a 0d 0a 09 74 65 6d 70 6c 61 rivate:...templa
0510: 74 65 20 3c 74 79 70 65 6e 61 6d 65 20 54 3e 20 te <typename T>
0520: 73 74 72 69 6e 67 20 70 72 69 6e 74 5f 61 72 72 string print_arr
0530: 61 79 28 63 6f 6e 73 74 20 76 65 63 74 6f 72 3c ay(const vector<
0540: 54 3e 20 26 56 29 20 7b 20 6f 73 74 72 69 6e 67 T> &V) { ostring
0550: 73 74 72 65 61 6d 20 6f 73 3b 20 6f 73 20 3c 3c stream os; os <<
0560: 20 22 7b 20 22 3b 20 66 6f 72 20 28 74 79 70 65 "{ "; for (type
0570: 6e 61 6d 65 20 76 65 63 74 6f 72 3c 54 3e 3a 3a name vector<T>::
0580: 63 6f 6e 73 74 5f 69 74 65 72 61 74 6f 72 20 69 const_iterator i
0590: 74 65 72 20 3d 20 56 2e 62 65 67 69 6e 28 29 3b ter = V.begin();
05a0: 20 69 74 65 72 20 21 3d 20 56 2e 65 6e 64 28 29 iter != V.end()
05b0: 3b 20 2b 2b 69 74 65 72 29 20 6f 73 20 3c 3c 20 ; ++iter) os <<
05c0: 27 5c 22 27 20 3c 3c 20 2a 69 74 65 72 20 3c 3c '\"' << *iter <<
05d0: 20 22 5c 22 2c 22 3b 20 6f 73 20 3c 3c 20 22 20 "\","; os << "
05e0: 7d 22 3b 20 72 65 74 75 72 6e 20 6f 73 2e 73 74 }"; return os.st
05f0: 72 28 29 3b 20 7d 0d 0a 09 76 6f 69 64 20 76 65 r(); }...void ve
0600: 72 69 66 79 5f 63 61 73 65 28 69 6e 74 20 43 61 rify_case(int Ca
0610: 73 65 2c 20 63 6f 6e 73 74 20 69 6e 74 20 26 45 se, const int &E
0620: 78 70 65 63 74 65 64 2c 20 63 6f 6e 73 74 20 69 xpected, const i
0630: 6e 74 20 26 52 65 63 65 69 76 65 64 29 20 7b 20 nt &Received) {
0640: 63 65 72 72 20 3c 3c 20 22 54 65 73 74 20 43 61 cerr << "Test Ca
0650: 73 65 20 23 22 20 3c 3c 20 43 61 73 65 20 3c 3c se #" << Case <<
0660: 20 22 2e 2e 2e 22 3b 20 69 66 20 28 45 78 70 65 "..."; if (Expe
0670: 63 74 65 64 20 3d 3d 20 52 65 63 65 69 76 65 64 cted == Received
0680: 29 20 63 65 72 72 20 3c 3c 20 22 50 41 53 53 45 ) cerr << "PASSE
0690: 44 22 20 3c 3c 20 65 6e 64 6c 3b 20 65 6c 73 65 D" << endl; else
06a0: 20 7b 20 63 65 72 72 20 3c 3c 20 22 46 41 49 4c { cerr << "FAIL
06b0: 45 44 22 20 3c 3c 20 65 6e 64 6c 3b 20 63 65 72 ED" << endl; cer
06c0: 72 20 3c 3c 20 22 5c 74 45 78 70 65 63 74 65 64 r << "\tExpected
06d0: 3a 20 5c 22 22 20 3c 3c 20 45 78 70 65 63 74 65 : \"" << Expecte
06e0: 64 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 65 6e 64 d << '\"' << end
06f0: 6c 3b 20 63 65 72 72 20 3c 3c 20 22 5c 74 52 65 l; cerr << "\tRe
0700: 63 65 69 76 65 64 3a 20 5c 22 22 20 3c 3c 20 52 ceived: \"" << R
0710: 65 63 65 69 76 65 64 20 3c 3c 20 27 5c 22 27 20 eceived << '\"'
0720: 3c 3c 20 65 6e 64 6c 3b 20 7d 20 7d 0d 0a 09 76 << endl; } }...v
0730: 6f 69 64 20 74 65 73 74 5f 63 61 73 65 5f 30 28 oid test_case_0(
0740: 29 20 7b 20 69 6e 74 20 41 72 72 30 5b 5d 20 3d ) { int Arr0[] =
0750: 20 7b 33 2c 32 7d 3b 20 76 65 63 74 6f 72 20 3c {3,2}; vector <
0760: 69 6e 74 3e 20 41 72 67 30 28 41 72 72 30 2c 20 int> Arg0(Arr0,
0770: 41 72 72 30 20 2b 20 28 73 69 7a 65 6f 66 28 41 Arr0 + (sizeof(A
0780: 72 72 30 29 20 2f 20 73 69 7a 65 6f 66 28 41 72 rr0) / sizeof(Ar
0790: 72 30 5b 30 5d 29 29 29 3b 20 69 6e 74 20 41 72 r0[0]))); int Ar
07a0: 72 31 5b 5d 20 3d 20 7b 31 2c 32 7d 3b 20 76 65 r1[] = {1,2}; ve
07b0: 63 74 6f 72 20 3c 69 6e 74 3e 20 41 72 67 31 28 ctor <int> Arg1(
07c0: 41 72 72 31 2c 20 41 72 72 31 20 2b 20 28 73 69 Arr1, Arr1 + (si
07d0: 7a 65 6f 66 28 41 72 72 31 29 20 2f 20 73 69 7a zeof(Arr1) / siz
07e0: 65 6f 66 28 41 72 72 31 5b 30 5d 29 29 29 3b 20 eof(Arr1[0])));
07f0: 69 6e 74 20 41 72 67 32 20 3d 20 35 3b 20 76 65 int Arg2 = 5; ve
0800: 72 69 66 79 5f 63 61 73 65 28 30 2c 20 41 72 67 rify_case(0, Arg
0810: 32 2c 20 67 65 74 53 63 6f 72 65 28 41 72 67 30 2, getScore(Arg0
0820: 2c 20 41 72 67 31 29 29 3b 20 7d 0d 0a 09 76 6f , Arg1)); }...vo
0830: 69 64 20 74 65 73 74 5f 63 61 73 65 5f 31 28 29 id test_case_1()
0840: 20 7b 20 69 6e 74 20 41 72 72 30 5b 5d 20 3d 20 { int Arr0[] =
0850: 7b 35 2c 34 2c 33 2c 32 2c 31 7d 3b 20 76 65 63 {5,4,3,2,1}; vec
0860: 74 6f 72 20 3c 69 6e 74 3e 20 41 72 67 30 28 41 tor <int> Arg0(A
0870: 72 72 30 2c 20 41 72 72 30 20 2b 20 28 73 69 7a rr0, Arr0 + (siz
0880: 65 6f 66 28 41 72 72 30 29 20 2f 20 73 69 7a 65 eof(Arr0) / size
0890: 6f 66 28 41 72 72 30 5b 30 5d 29 29 29 3b 20 69 of(Arr0[0]))); i
08a0: 6e 74 20 41 72 72 31 5b 5d 20 3d 20 7b 31 2c 31 nt Arr1[] = {1,1
08b0: 2c 31 2c 31 2c 31 7d 3b 20 76 65 63 74 6f 72 20 ,1,1,1}; vector
08c0: 3c 69 6e 74 3e 20 41 72 67 31 28 41 72 72 31 2c <int> Arg1(Arr1,
08d0: 20 41 72 72 31 20 2b 20 28 73 69 7a 65 6f 66 28 Arr1 + (sizeof(
08e0: 41 72 72 31 29 20 2f 20 73 69 7a 65 6f 66 28 41 Arr1) / sizeof(A
08f0: 72 72 31 5b 30 5d 29 29 29 3b 20 69 6e 74 20 41 rr1[0]))); int A
0900: 72 67 32 20 3d 20 39 3b 20 76 65 72 69 66 79 5f rg2 = 9; verify_
0910: 63 61 73 65 28 31 2c 20 41 72 67 32 2c 20 67 65 case(1, Arg2, ge
0920: 74 53 63 6f 72 65 28 41 72 67 30 2c 20 41 72 67 tScore(Arg0, Arg
0930: 31 29 29 3b 20 7d 0d 0a 09 76 6f 69 64 20 74 65 1)); }...void te
0940: 73 74 5f 63 61 73 65 5f 32 28 29 20 7b 20 69 6e st_case_2() { in
0950: 74 20 41 72 72 30 5b 5d 20 3d 20 7b 35 2c 35 7d t Arr0[] = {5,5}
0960: 3b 20 76 65 63 74 6f 72 20 3c 69 6e 74 3e 20 41 ; vector <int> A
0970: 72 67 30 28 41 72 72 30 2c 20 41 72 72 30 20 2b rg0(Arr0, Arr0 +
0980: 20 28 73 69 7a 65 6f 66 28 41 72 72 30 29 20 2f (sizeof(Arr0) /
0990: 20 73 69 7a 65 6f 66 28 41 72 72 30 5b 30 5d 29 sizeof(Arr0[0])
09a0: 29 29 3b 20 69 6e 74 20 41 72 72 31 5b 5d 20 3d )); int Arr1[] =
09b0: 20 7b 32 2c 32 7d 3b 20 76 65 63 74 6f 72 20 3c {2,2}; vector <
09c0: 69 6e 74 3e 20 41 72 67 31 28 41 72 72 31 2c 20 int> Arg1(Arr1,
09d0: 41 72 72 31 20 2b 20 28 73 69 7a 65 6f 66 28 41 Arr1 + (sizeof(A
09e0: 72 72 31 29 20 2f 20 73 69 7a 65 6f 66 28 41 72 rr1) / sizeof(Ar
09f0: 72 31 5b 30 5d 29 29 29 3b 20 69 6e 74 20 41 72 r1[0]))); int Ar
0a00: 67 32 20 3d 20 30 3b 20 76 65 72 69 66 79 5f 63 g2 = 0; verify_c
0a10: 61 73 65 28 32 2c 20 41 72 67 32 2c 20 67 65 74 ase(2, Arg2, get
0a20: 53 63 6f 72 65 28 41 72 67 30 2c 20 41 72 67 31 Score(Arg0, Arg1
0a30: 29 29 3b 20 7d 0d 0a 09 76 6f 69 64 20 74 65 73 )); }...void tes
0a40: 74 5f 63 61 73 65 5f 33 28 29 20 7b 20 69 6e 74 t_case_3() { int
0a50: 20 41 72 72 30 5b 5d 20 3d 20 7b 31 7d 3b 20 76 Arr0[] = {1}; v
0a60: 65 63 74 6f 72 20 3c 69 6e 74 3e 20 41 72 67 30 ector <int> Arg0
0a70: 28 41 72 72 30 2c 20 41 72 72 30 20 2b 20 28 73 (Arr0, Arr0 + (s
0a80: 69 7a 65 6f 66 28 41 72 72 30 29 20 2f 20 73 69 izeof(Arr0) / si
0a90: 7a 65 6f 66 28 41 72 72 30 5b 30 5d 29 29 29 3b zeof(Arr0[0])));
0aa0: 20 69 6e 74 20 41 72 72 31 5b 5d 20 3d 20 7b 31 int Arr1[] = {1
0ab0: 30 7d 3b 20 76 65 63 74 6f 72 20 3c 69 6e 74 3e 0}; vector <int>
0ac0: 20 41 72 67 31 28 41 72 72 31 2c 20 41 72 72 31 Arg1(Arr1, Arr1
0ad0: 20 2b 20 28 73 69 7a 65 6f 66 28 41 72 72 31 29 + (sizeof(Arr1)
0ae0: 20 2f 20 73 69 7a 65 6f 66 28 41 72 72 31 5b 30 / sizeof(Arr1[0
0af0: 5d 29 29 29 3b 20 69 6e 74 20 41 72 67 32 20 3d ]))); int Arg2 =
0b00: 20 30 3b 20 76 65 72 69 66 79 5f 63 61 73 65 28 0; verify_case(
0b10: 33 2c 20 41 72 67 32 2c 20 67 65 74 53 63 6f 72 3, Arg2, getScor
0b20: 65 28 41 72 67 30 2c 20 41 72 67 31 29 29 3b 20 e(Arg0, Arg1));
0b30: 7d 0d 0a 0d 0a 2f 2f 20 45 4e 44 20 43 55 54 20 }....// END CUT
0b40: 48 45 52 45 0d 0a 7d 3b 0d 0a 2f 2f 20 42 45 47 HERE..};..// BEG
0b50: 49 4e 20 43 55 54 20 48 45 52 45 20 0d 0a 69 6e IN CUT HERE ..in
0b60: 74 20 6d 61 69 6e 28 29 20 7b 20 53 74 6f 6e 65 t main() { Stone
0b70: 47 61 6d 65 28 29 2e 72 75 6e 5f 74 65 73 74 28 Game().run_test(
0b80: 2d 31 29 3b 20 7d 0d 0a 2f 2f 20 45 4e 44 20 43 -1); }..// END C
0b90: 55 54 20 48 45 52 45 0d 0a UT HERE..