Artifact 6420102b826df671b84826ae5d107117220ef3b6:
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 23 69 66 64 65 <cstring>..#ifde
0150: 66 20 5f 5f 47 4e 55 43 5f 5f 0d 0a 23 69 6e 63 f __GNUC__..#inc
0160: 6c 75 64 65 20 3c 65 78 74 2f 68 61 73 68 5f 6d lude <ext/hash_m
0170: 61 70 3e 0d 0a 23 64 65 66 69 6e 65 20 75 6e 6f ap>..#define uno
0180: 72 64 65 72 65 64 5f 6d 61 70 20 5f 5f 67 6e 75 rdered_map __gnu
0190: 5f 63 78 78 3a 3a 68 61 73 68 5f 6d 61 70 0d 0a _cxx::hash_map..
01a0: 23 65 6c 73 65 0d 0a 23 69 6e 63 6c 75 64 65 20 #else..#include
01b0: 3c 75 6e 6f 72 64 65 72 65 64 5f 6d 61 70 3e 0d <unordered_map>.
01c0: 0a 23 65 6e 64 69 66 0d 0a 75 73 69 6e 67 20 6e .#endif..using n
01d0: 61 6d 65 73 70 61 63 65 20 73 74 64 3b 0d 0a 74 amespace std;..t
01e0: 79 70 65 64 65 66 20 6c 6f 6e 67 20 6c 6f 6e 67 ypedef long long
01f0: 20 4c 4c 3b 0d 0a 74 79 70 65 64 65 66 20 63 6f LL;..typedef co
0200: 6d 70 6c 65 78 3c 64 6f 75 62 6c 65 3e 20 43 4d mplex<double> CM
0210: 50 3b 0d 0a 73 74 61 74 69 63 20 63 6f 6e 73 74 P;..static const
0220: 20 69 6e 74 20 49 4e 46 20 3d 20 30 78 33 66 66 int INF = 0x3ff
0230: 66 66 66 66 66 3b 0d 0a 0d 0a 63 6c 61 73 73 20 fffff;....class
0240: 50 72 69 6d 65 43 6f 6d 70 6f 73 69 74 65 47 61 PrimeCompositeGa
0250: 6d 65 20 7b 20 70 75 62 6c 69 63 3a 0d 0a 09 69 me { public:...i
0260: 6e 74 20 74 68 65 4f 75 74 63 6f 6d 65 28 69 6e nt theOutcome(in
0270: 74 20 4e 2c 20 69 6e 74 20 4b 29 0d 0a 09 7b 0d t N, int K)...{.
0280: 0a 09 09 65 6e 75 6d 20 7b 50 2c 20 43 7d 3b 0d ...enum {P, C};.
0290: 0a 0d 0a 09 09 2f 2f 20 73 69 65 76 65 0d 0a 09 .....// sieve...
02a0: 09 76 65 63 74 6f 72 3c 20 76 65 63 74 6f 72 3c .vector< vector<
02b0: 62 6f 6f 6c 3e 20 3e 20 69 73 28 32 2c 20 76 65 bool> > is(2, ve
02c0: 63 74 6f 72 3c 62 6f 6f 6c 3e 28 4e 2b 31 2c 20 ctor<bool>(N+1,
02d0: 74 72 75 65 29 29 3b 0d 0a 09 09 69 73 5b 50 5d true));....is[P]
02e0: 5b 30 5d 20 3d 20 69 73 5b 43 5d 5b 30 5d 20 3d [0] = is[C][0] =
02f0: 20 69 73 5b 50 5d 5b 31 5d 20 3d 20 69 73 5b 43 is[P][1] = is[C
0300: 5d 5b 31 5d 20 3d 20 66 61 6c 73 65 3b 0d 0a 09 ][1] = false;...
0310: 09 66 6f 72 28 69 6e 74 20 70 3d 32 3b 20 70 3c .for(int p=2; p<
0320: 3d 4e 3b 20 2b 2b 70 29 20 7b 0d 0a 09 09 09 69 =N; ++p) {.....i
0330: 66 28 20 69 73 5b 50 5d 5b 70 5d 20 29 0d 0a 09 f( is[P][p] )...
0340: 09 09 09 66 6f 72 28 69 6e 74 20 71 3d 70 2b 70 ...for(int q=p+p
0350: 3b 20 71 3c 3d 4e 3b 20 71 2b 3d 70 29 0d 0a 09 ; q<=N; q+=p)...
0360: 09 09 09 09 69 73 5b 50 5d 5b 71 5d 20 3d 20 66 ....is[P][q] = f
0370: 61 6c 73 65 3b 0d 0a 09 09 09 69 73 5b 43 5d 5b alse;.....is[C][
0380: 70 5d 20 3d 20 21 69 73 5b 50 5d 5b 70 5d 3b 0d p] = !is[P][p];.
0390: 0a 09 09 7d 0d 0a 0d 0a 09 09 76 65 63 74 6f 72 ...}......vector
03a0: 3c 20 76 65 63 74 6f 72 3c 69 6e 74 3e 20 3e 20 < vector<int> >
03b0: 73 63 6f 72 65 28 32 2c 20 76 65 63 74 6f 72 3c score(2, vector<
03c0: 69 6e 74 3e 28 4e 2b 31 29 29 3b 0d 0a 09 09 6d int>(N+1));....m
03d0: 75 6c 74 69 73 65 74 3c 69 6e 74 3e 20 6c 61 73 ultiset<int> las
03e0: 74 4b 5b 32 5d 3b 0d 0a 09 09 66 6f 72 28 69 6e tK[2];....for(in
03f0: 74 20 74 3d 30 3b 20 74 3c 3d 4e 3b 20 2b 2b 74 t t=0; t<=N; ++t
0400: 29 20 7b 0d 0a 09 09 09 66 6f 72 28 69 6e 74 20 ) {.....for(int
0410: 6d 65 3d 30 3b 20 6d 65 3c 32 3b 20 2b 2b 6d 65 me=0; me<2; ++me
0420: 29 0d 0a 09 09 09 09 69 66 28 20 6c 61 73 74 4b )......if( lastK
0430: 5b 31 5e 6d 65 5d 2e 65 6d 70 74 79 28 29 20 29 [1^me].empty() )
0440: 20 2f 2f 20 6e 6f 20 76 61 6c 69 64 20 6d 6f 76 // no valid mov
0450: 65 2e 20 69 6d 6d 65 64 69 61 74 65 6c 79 20 6c e. immediately l
0460: 6f 73 65 2e 0d 0a 09 09 09 09 09 73 63 6f 72 65 ose........score
0470: 5b 6d 65 5d 5b 74 5d 20 3d 20 30 3b 0d 0a 09 09 [me][t] = 0;....
0480: 09 09 65 6c 73 65 20 69 66 28 20 2a 6c 61 73 74 ..else if( *last
0490: 4b 5b 31 5e 6d 65 5d 2e 62 65 67 69 6e 28 29 20 K[1^me].begin()
04a0: 3e 20 30 20 29 20 2f 2f 20 61 6c 6c 20 6d 6f 76 > 0 ) // all mov
04b0: 65 20 6d 61 6b 65 20 6f 70 70 6f 6e 65 6e 74 20 e make opponent
04c0: 77 69 6e 2e 20 74 61 6b 65 20 74 68 65 20 6c 6f win. take the lo
04d0: 6e 67 65 73 74 2e 0d 0a 09 09 09 09 09 73 63 6f ngest........sco
04e0: 72 65 5b 6d 65 5d 5b 74 5d 20 3d 20 2d 20 28 31 re[me][t] = - (1
04f0: 20 2b 20 2a 6c 61 73 74 4b 5b 31 5e 6d 65 5d 2e + *lastK[1^me].
0500: 72 62 65 67 69 6e 28 29 29 3b 0d 0a 09 09 09 09 rbegin());......
0510: 65 6c 73 65 20 2f 2f 20 49 20 63 61 6e 20 77 69 else // I can wi
0520: 6e 2e 20 74 61 6b 65 20 74 68 65 20 73 68 6f 72 n. take the shor
0530: 74 65 73 74 2e 0d 0a 09 09 09 09 09 73 63 6f 72 test........scor
0540: 65 5b 6d 65 5d 5b 74 5d 20 3d 20 28 31 20 2b 20 e[me][t] = (1 +
0550: 61 62 73 28 2a 2d 2d 6c 61 73 74 4b 5b 31 5e 6d abs(*--lastK[1^m
0560: 65 5d 2e 6c 6f 77 65 72 5f 62 6f 75 6e 64 28 31 e].lower_bound(1
0570: 29 29 29 3b 0d 0a 09 09 09 2f 2f 20 6d 61 69 6e )));.....// main
0580: 74 61 69 6e 20 6c 61 73 74 20 4b 20 73 63 6f 72 tain last K scor
0590: 65 73 2e 0d 0a 09 09 09 66 6f 72 28 69 6e 74 20 es......for(int
05a0: 6d 65 3d 30 3b 20 6d 65 3c 32 3b 20 2b 2b 6d 65 me=0; me<2; ++me
05b0: 29 20 7b 0d 0a 09 09 09 09 69 66 28 20 69 73 5b ) {......if( is[
05c0: 31 5e 6d 65 5d 5b 74 5d 20 29 0d 0a 09 09 09 09 1^me][t] )......
05d0: 09 6c 61 73 74 4b 5b 6d 65 5d 2e 69 6e 73 65 72 .lastK[me].inser
05e0: 74 28 20 73 63 6f 72 65 5b 6d 65 5d 5b 74 5d 20 t( score[me][t]
05f0: 29 3b 0d 0a 09 09 09 09 69 66 28 20 74 2d 4b 3e );......if( t-K>
0600: 3d 30 20 26 26 20 69 73 5b 31 5e 6d 65 5d 5b 74 =0 && is[1^me][t
0610: 2d 4b 5d 20 29 0d 0a 09 09 09 09 09 6c 61 73 74 -K] ).......last
0620: 4b 5b 6d 65 5d 2e 65 72 61 73 65 28 6c 61 73 74 K[me].erase(last
0630: 4b 5b 6d 65 5d 2e 66 69 6e 64 28 73 63 6f 72 65 K[me].find(score
0640: 5b 6d 65 5d 5b 74 2d 4b 5d 29 29 3b 0d 0a 09 09 [me][t-K]));....
0650: 09 7d 0d 0a 09 09 7d 0d 0a 09 09 72 65 74 75 72 .}....}....retur
0660: 6e 20 73 63 6f 72 65 5b 50 5d 5b 4e 5d 3b 0d 0a n score[P][N];..
0670: 09 7d 0d 0a 7d 3b 0d 0a 0d 0a 2f 2f 20 42 45 47 .}..};....// BEG
0680: 49 4e 20 43 55 54 20 48 45 52 45 0d 0a 23 69 6e IN CUT HERE..#in
0690: 63 6c 75 64 65 20 3c 63 74 69 6d 65 3e 0d 0a 64 clude <ctime>..d
06a0: 6f 75 62 6c 65 20 73 74 61 72 74 5f 74 69 6d 65 ouble start_time
06b0: 3b 20 73 74 72 69 6e 67 20 74 69 6d 65 72 28 29 ; string timer()
06c0: 0d 0a 20 7b 20 6f 73 74 72 69 6e 67 73 74 72 65 .. { ostringstre
06d0: 61 6d 20 6f 73 3b 20 6f 73 20 3c 3c 20 22 20 28 am os; os << " (
06e0: 22 20 3c 3c 20 69 6e 74 28 28 63 6c 6f 63 6b 28 " << int((clock(
06f0: 29 2d 73 74 61 72 74 5f 74 69 6d 65 29 2f 43 4c )-start_time)/CL
0700: 4f 43 4b 53 5f 50 45 52 5f 53 45 43 2a 31 30 30 OCKS_PER_SEC*100
0710: 30 29 20 3c 3c 20 22 20 6d 73 65 63 29 22 3b 20 0) << " msec)";
0720: 72 65 74 75 72 6e 20 6f 73 2e 73 74 72 28 29 3b return os.str();
0730: 20 7d 0d 0a 74 65 6d 70 6c 61 74 65 3c 74 79 70 }..template<typ
0740: 65 6e 61 6d 65 20 54 3e 20 6f 73 74 72 65 61 6d ename T> ostream
0750: 26 20 6f 70 65 72 61 74 6f 72 3c 3c 28 6f 73 74 & operator<<(ost
0760: 72 65 61 6d 26 20 6f 73 2c 20 63 6f 6e 73 74 20 ream& os, const
0770: 76 65 63 74 6f 72 3c 54 3e 26 20 76 29 0d 0a 20 vector<T>& v)..
0780: 7b 20 6f 73 20 3c 3c 20 22 7b 20 22 3b 0d 0a 20 { os << "{ ";..
0790: 20 20 66 6f 72 28 74 79 70 65 6e 61 6d 65 20 76 for(typename v
07a0: 65 63 74 6f 72 3c 54 3e 3a 3a 63 6f 6e 73 74 5f ector<T>::const_
07b0: 69 74 65 72 61 74 6f 72 20 69 74 3d 76 2e 62 65 iterator it=v.be
07c0: 67 69 6e 28 29 3b 20 69 74 21 3d 76 2e 65 6e 64 gin(); it!=v.end
07d0: 28 29 3b 20 2b 2b 69 74 29 0d 0a 20 20 20 6f 73 (); ++it).. os
07e0: 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 2a 69 74 20 << '\"' << *it
07f0: 3c 3c 20 27 5c 22 27 20 3c 3c 20 28 69 74 2b 31 << '\"' << (it+1
0800: 3d 3d 76 2e 65 6e 64 28 29 20 3f 20 22 22 20 3a ==v.end() ? "" :
0810: 20 22 2c 20 22 29 3b 20 6f 73 20 3c 3c 20 22 20 ", "); os << "
0820: 7d 22 3b 20 72 65 74 75 72 6e 20 6f 73 3b 20 7d }"; return os; }
0830: 0d 0a 76 6f 69 64 20 76 65 72 69 66 79 5f 63 61 ..void verify_ca
0840: 73 65 28 63 6f 6e 73 74 20 69 6e 74 26 20 45 78 se(const int& Ex
0850: 70 65 63 74 65 64 2c 20 63 6f 6e 73 74 20 69 6e pected, const in
0860: 74 26 20 52 65 63 65 69 76 65 64 29 20 7b 0d 0a t& Received) {..
0870: 20 62 6f 6f 6c 20 6f 6b 20 3d 20 28 45 78 70 65 bool ok = (Expe
0880: 63 74 65 64 20 3d 3d 20 52 65 63 65 69 76 65 64 cted == Received
0890: 29 3b 0d 0a 20 69 66 28 6f 6b 29 20 63 65 72 72 );.. if(ok) cerr
08a0: 20 3c 3c 20 22 50 41 53 53 45 44 22 20 3c 3c 20 << "PASSED" <<
08b0: 74 69 6d 65 72 28 29 20 3c 3c 20 65 6e 64 6c 3b timer() << endl;
08c0: 20 20 65 6c 73 65 20 7b 20 63 65 72 72 20 3c 3c else { cerr <<
08d0: 20 22 46 41 49 4c 45 44 22 20 3c 3c 20 74 69 6d "FAILED" << tim
08e0: 65 72 28 29 20 3c 3c 20 65 6e 64 6c 3b 0d 0a 20 er() << endl;..
08f0: 63 65 72 72 20 3c 3c 20 22 5c 74 6f 3a 20 5c 22 cerr << "\to: \"
0900: 22 20 3c 3c 20 45 78 70 65 63 74 65 64 20 3c 3c " << Expected <<
0910: 20 27 5c 22 27 20 3c 3c 20 65 6e 64 6c 20 3c 3c '\"' << endl <<
0920: 20 22 5c 74 78 3a 20 5c 22 22 20 3c 3c 20 52 65 "\tx: \"" << Re
0930: 63 65 69 76 65 64 20 3c 3c 20 27 5c 22 27 20 3c ceived << '\"' <
0940: 3c 20 65 6e 64 6c 3b 20 7d 20 7d 0d 0a 23 64 65 < endl; } }..#de
0950: 66 69 6e 65 20 43 41 53 45 28 4e 29 20 7b 63 65 fine CASE(N) {ce
0960: 72 72 20 3c 3c 20 22 54 65 73 74 20 43 61 73 65 rr << "Test Case
0970: 20 23 22 20 3c 3c 20 4e 20 3c 3c 20 22 2e 2e 2e #" << N << "...
0980: 22 20 3c 3c 20 66 6c 75 73 68 3b 20 73 74 61 72 " << flush; star
0990: 74 5f 74 69 6d 65 3d 63 6c 6f 63 6b 28 29 3b 0d t_time=clock();.
09a0: 0a 23 64 65 66 69 6e 65 20 45 4e 44 09 20 76 65 .#define END. ve
09b0: 72 69 66 79 5f 63 61 73 65 28 5f 2c 20 50 72 69 rify_case(_, Pri
09c0: 6d 65 43 6f 6d 70 6f 73 69 74 65 47 61 6d 65 28 meCompositeGame(
09d0: 29 2e 74 68 65 4f 75 74 63 6f 6d 65 28 4e 2c 20 ).theOutcome(N,
09e0: 4b 29 29 3b 7d 0d 0a 69 6e 74 20 6d 61 69 6e 28 K));}..int main(
09f0: 29 7b 0d 0a 0d 0a 43 41 53 45 28 30 29 0d 0a 09 ){....CASE(0)...
0a00: 69 6e 74 20 4e 20 3d 20 33 3b 20 0d 0a 09 69 6e int N = 3; ...in
0a10: 74 20 4b 20 3d 20 32 3b 20 0d 0a 09 69 6e 74 20 t K = 2; ...int
0a20: 5f 20 3d 20 31 3b 20 0d 0a 45 4e 44 0d 0a 43 41 _ = 1; ..END..CA
0a30: 53 45 28 31 29 0d 0a 09 69 6e 74 20 4e 20 3d 20 SE(1)...int N =
0a40: 35 38 3b 20 0d 0a 09 69 6e 74 20 4b 20 3d 20 31 58; ...int K = 1
0a50: 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 30 3b 20 ; ...int _ = 0;
0a60: 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 32 29 0d 0a ..END..CASE(2)..
0a70: 09 69 6e 74 20 4e 20 3d 20 33 30 3b 20 0d 0a 09 .int N = 30; ...
0a80: 69 6e 74 20 4b 20 3d 20 33 3b 20 0d 0a 09 69 6e int K = 3; ...in
0a90: 74 20 5f 20 3d 20 2d 32 3b 20 0d 0a 45 4e 44 0d t _ = -2; ..END.
0aa0: 0a 43 41 53 45 28 33 29 0d 0a 09 69 6e 74 20 4e .CASE(3)...int N
0ab0: 20 3d 20 36 3b 20 0d 0a 09 69 6e 74 20 4b 20 3d = 6; ...int K =
0ac0: 20 33 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 31 3; ...int _ = 1
0ad0: 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 34 29 ; ..END..CASE(4)
0ae0: 0d 0a 09 69 6e 74 20 4e 20 3d 20 35 32 36 3b 20 ...int N = 526;
0af0: 0d 0a 09 69 6e 74 20 4b 20 3d 20 35 38 3b 20 0d ...int K = 58; .
0b00: 0a 09 69 6e 74 20 5f 20 3d 20 31 39 3b 20 0d 0a ..int _ = 19; ..
0b10: 45 4e 44 0d 0a 43 41 53 45 28 35 29 0d 0a 09 69 END..CASE(5)...i
0b20: 6e 74 20 4e 20 3d 20 34 37 34 37 34 37 3b 20 0d nt N = 474747; .
0b30: 0a 09 69 6e 74 20 4b 20 3d 20 34 37 34 37 34 37 ..int K = 474747
0b40: 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 31 3b 20 ; ...int _ = 1;
0b50: 0d 0a 45 4e 44 0d 0a 2f 2a 0d 0a 43 41 53 45 28 ..END../*..CASE(
0b60: 36 29 0d 0a 09 69 6e 74 20 4e 20 3d 20 3b 20 0d 6)...int N = ; .
0b70: 0a 09 69 6e 74 20 4b 20 3d 20 3b 20 0d 0a 09 69 ..int K = ; ...i
0b80: 6e 74 20 5f 20 3d 20 3b 20 0d 0a 45 4e 44 0d 0a nt _ = ; ..END..
0b90: 2a 2f 0d 0a 7d 0d 0a 2f 2f 20 45 4e 44 20 43 55 */..}..// END CU
0ba0: 54 20 48 45 52 45 0d 0a T HERE..