Artifact 0939e8f38eb262b66ebd196a5505d55379c4d7e8:
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 5a 6f CMP;....class Zo
01a0: 6f 20 7b 20 70 75 62 6c 69 63 3a 0d 0a 09 6c 6f o { public:...lo
01b0: 6e 67 20 6c 6f 6e 67 20 74 68 65 43 6f 75 6e 74 ng long theCount
01c0: 28 76 65 63 74 6f 72 20 3c 69 6e 74 3e 20 61 6e (vector <int> an
01d0: 73 77 65 72 73 29 20 0d 0a 09 7b 0d 0a 09 09 6d swers) ...{....m
01e0: 61 70 3c 69 6e 74 2c 69 6e 74 3e 20 6e 63 3b 0d ap<int,int> nc;.
01f0: 0a 09 09 66 6f 72 28 69 6e 74 20 69 3d 30 3b 20 ...for(int i=0;
0200: 69 3c 61 6e 73 77 65 72 73 2e 73 69 7a 65 28 29 i<answers.size()
0210: 3b 20 2b 2b 69 29 0d 0a 09 09 09 6e 63 5b 61 6e ; ++i).....nc[an
0220: 73 77 65 72 73 5b 69 5d 5d 20 2b 2b 3b 0d 0a 0d swers[i]] ++;...
0230: 0a 09 09 69 6e 74 20 74 77 6f 20 3d 20 30 3b 0d ...int two = 0;.
0240: 0a 09 09 69 6e 74 20 6f 6e 65 20 3d 20 30 3b 0d ...int one = 0;.
0250: 0a 09 09 66 6f 72 28 69 6e 74 20 6b 3d 30 20 3b ...for(int k=0 ;
0260: 3b 20 2b 2b 6b 29 0d 0a 09 09 09 69 66 28 20 6e ; ++k).....if( n
0270: 63 5b 6b 5d 20 3d 3d 20 32 20 29 0d 0a 09 09 09 c[k] == 2 ).....
0280: 09 74 77 6f 2b 2b 3b 0d 0a 09 09 09 65 6c 73 65 .two++;.....else
0290: 20 69 66 28 20 6e 63 5b 6b 5d 20 3d 3d 20 31 20 if( nc[k] == 1
02a0: 29 0d 0a 09 09 09 7b 0d 0a 09 09 09 09 66 6f 72 ).....{......for
02b0: 28 3b 3b 20 2b 2b 6b 29 0d 0a 09 09 09 09 09 69 (;; ++k).......i
02c0: 66 28 20 6e 63 5b 6b 5d 20 3d 3d 20 31 20 29 0d f( nc[k] == 1 ).
02d0: 0a 09 09 09 09 09 09 6f 6e 65 2b 2b 3b 0d 0a 09 .......one++;...
02e0: 09 09 09 09 65 6c 73 65 20 69 66 28 20 6e 63 5b ....else if( nc[
02f0: 6b 5d 20 3d 3d 20 30 20 29 0d 0a 09 09 09 09 09 k] == 0 ).......
0300: 09 62 72 65 61 6b 3b 0d 0a 09 09 09 09 09 65 6c .break;.......el
0310: 73 65 0d 0a 09 09 09 09 09 09 72 65 74 75 72 6e se........return
0320: 20 30 4c 4c 3b 0d 0a 09 09 09 09 62 72 65 61 6b 0LL;......break
0330: 3b 0d 0a 09 09 09 7d 0d 0a 09 09 09 65 6c 73 65 ;.....}.....else
0340: 20 69 66 28 20 6e 63 5b 6b 5d 20 3d 3d 20 30 20 if( nc[k] == 0
0350: 29 0d 0a 09 09 09 09 62 72 65 61 6b 3b 0d 0a 09 )......break;...
0360: 09 09 65 6c 73 65 0d 0a 09 09 09 09 72 65 74 75 ..else......retu
0370: 72 6e 20 30 4c 4c 3b 0d 0a 09 09 69 66 28 20 74 rn 0LL;....if( t
0380: 77 6f 2a 32 20 2b 20 6f 6e 65 20 21 3d 20 61 6e wo*2 + one != an
0390: 73 77 65 72 73 2e 73 69 7a 65 28 29 20 29 0d 0a swers.size() )..
03a0: 09 09 09 72 65 74 75 72 6e 20 30 4c 4c 3b 0d 0a ...return 0LL;..
03b0: 09 09 4c 4c 20 72 20 3d 20 32 3b 0d 0a 09 09 66 ..LL r = 2;....f
03c0: 6f 72 28 69 6e 74 20 69 3d 30 3b 20 69 3c 74 77 or(int i=0; i<tw
03d0: 6f 3b 20 2b 2b 69 29 0d 0a 09 09 09 72 20 3d 20 o; ++i).....r =
03e0: 72 2a 32 3b 0d 0a 09 09 72 65 74 75 72 6e 20 72 r*2;....return r
03f0: 3b 0d 0a 09 7d 0d 0a 7d 3b 0d 0a 0d 0a 2f 2f 20 ;...}..};....//
0400: 42 45 47 49 4e 20 43 55 54 20 48 45 52 45 0d 0a BEGIN CUT HERE..
0410: 23 69 6e 63 6c 75 64 65 20 3c 63 74 69 6d 65 3e #include <ctime>
0420: 0d 0a 64 6f 75 62 6c 65 20 73 74 61 72 74 5f 74 ..double start_t
0430: 69 6d 65 3b 20 73 74 72 69 6e 67 20 74 69 6d 65 ime; string time
0440: 72 28 29 0d 0a 20 7b 20 6f 73 74 72 69 6e 67 73 r().. { ostrings
0450: 74 72 65 61 6d 20 6f 73 3b 20 6f 73 20 3c 3c 20 tream os; os <<
0460: 22 20 28 22 20 3c 3c 20 69 6e 74 28 28 63 6c 6f " (" << int((clo
0470: 63 6b 28 29 2d 73 74 61 72 74 5f 74 69 6d 65 29 ck()-start_time)
0480: 2f 43 4c 4f 43 4b 53 5f 50 45 52 5f 53 45 43 2a /CLOCKS_PER_SEC*
0490: 31 30 30 30 29 20 3c 3c 20 22 20 6d 73 65 63 29 1000) << " msec)
04a0: 22 3b 20 72 65 74 75 72 6e 20 6f 73 2e 73 74 72 "; return os.str
04b0: 28 29 3b 20 7d 0d 0a 74 65 6d 70 6c 61 74 65 3c (); }..template<
04c0: 74 79 70 65 6e 61 6d 65 20 54 3e 20 6f 73 74 72 typename T> ostr
04d0: 65 61 6d 26 20 6f 70 65 72 61 74 6f 72 3c 3c 28 eam& operator<<(
04e0: 6f 73 74 72 65 61 6d 26 20 6f 73 2c 20 63 6f 6e ostream& os, con
04f0: 73 74 20 76 65 63 74 6f 72 3c 54 3e 26 20 76 29 st vector<T>& v)
0500: 0d 0a 20 7b 20 6f 73 20 3c 3c 20 22 7b 20 22 3b .. { os << "{ ";
0510: 0d 0a 20 20 20 66 6f 72 28 74 79 70 65 6e 61 6d .. for(typenam
0520: 65 20 76 65 63 74 6f 72 3c 54 3e 3a 3a 63 6f 6e e vector<T>::con
0530: 73 74 5f 69 74 65 72 61 74 6f 72 20 69 74 3d 76 st_iterator it=v
0540: 2e 62 65 67 69 6e 28 29 3b 20 69 74 21 3d 76 2e .begin(); it!=v.
0550: 65 6e 64 28 29 3b 20 2b 2b 69 74 29 0d 0a 20 20 end(); ++it)..
0560: 20 6f 73 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 2a os << '\"' << *
0570: 69 74 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 28 69 it << '\"' << (i
0580: 74 2b 31 3d 3d 76 2e 65 6e 64 28 29 20 3f 20 22 t+1==v.end() ? "
0590: 22 20 3a 20 22 2c 20 22 29 3b 20 6f 73 20 3c 3c " : ", "); os <<
05a0: 20 22 20 7d 22 3b 20 72 65 74 75 72 6e 20 6f 73 " }"; return os
05b0: 3b 20 7d 0d 0a 76 6f 69 64 20 76 65 72 69 66 79 ; }..void verify
05c0: 5f 63 61 73 65 28 63 6f 6e 73 74 20 6c 6f 6e 67 _case(const long
05d0: 20 6c 6f 6e 67 26 20 45 78 70 65 63 74 65 64 2c long& Expected,
05e0: 20 63 6f 6e 73 74 20 6c 6f 6e 67 20 6c 6f 6e 67 const long long
05f0: 26 20 52 65 63 65 69 76 65 64 29 20 7b 0d 0a 20 & Received) {..
0600: 62 6f 6f 6c 20 6f 6b 20 3d 20 28 45 78 70 65 63 bool ok = (Expec
0610: 74 65 64 20 3d 3d 20 52 65 63 65 69 76 65 64 29 ted == Received)
0620: 3b 0d 0a 20 69 66 28 6f 6b 29 20 63 65 72 72 20 ;.. if(ok) cerr
0630: 3c 3c 20 22 50 41 53 53 45 44 22 20 3c 3c 20 74 << "PASSED" << t
0640: 69 6d 65 72 28 29 20 3c 3c 20 65 6e 64 6c 3b 20 imer() << endl;
0650: 20 65 6c 73 65 20 7b 20 63 65 72 72 20 3c 3c 20 else { cerr <<
0660: 22 46 41 49 4c 45 44 22 20 3c 3c 20 74 69 6d 65 "FAILED" << time
0670: 72 28 29 20 3c 3c 20 65 6e 64 6c 3b 0d 0a 20 63 r() << endl;.. c
0680: 65 72 72 20 3c 3c 20 22 5c 74 6f 3a 20 5c 22 22 err << "\to: \""
0690: 20 3c 3c 20 45 78 70 65 63 74 65 64 20 3c 3c 20 << Expected <<
06a0: 27 5c 22 27 20 3c 3c 20 65 6e 64 6c 20 3c 3c 20 '\"' << endl <<
06b0: 22 5c 74 78 3a 20 5c 22 22 20 3c 3c 20 52 65 63 "\tx: \"" << Rec
06c0: 65 69 76 65 64 20 3c 3c 20 27 5c 22 27 20 3c 3c eived << '\"' <<
06d0: 20 65 6e 64 6c 3b 20 7d 20 7d 0d 0a 23 64 65 66 endl; } }..#def
06e0: 69 6e 65 20 43 41 53 45 28 4e 29 20 7b 63 65 72 ine CASE(N) {cer
06f0: 72 20 3c 3c 20 22 54 65 73 74 20 43 61 73 65 20 r << "Test Case
0700: 23 22 20 3c 3c 20 4e 20 3c 3c 20 22 2e 2e 2e 22 #" << N << "..."
0710: 20 3c 3c 20 66 6c 75 73 68 3b 20 73 74 61 72 74 << flush; start
0720: 5f 74 69 6d 65 3d 63 6c 6f 63 6b 28 29 3b 0d 0a _time=clock();..
0730: 23 64 65 66 69 6e 65 20 45 4e 44 09 20 76 65 72 #define END. ver
0740: 69 66 79 5f 63 61 73 65 28 5f 2c 20 5a 6f 6f 28 ify_case(_, Zoo(
0750: 29 2e 74 68 65 43 6f 75 6e 74 28 61 6e 73 77 65 ).theCount(answe
0760: 72 73 29 29 3b 7d 0d 0a 69 6e 74 20 6d 61 69 6e rs));}..int main
0770: 28 29 7b 0d 0a 0d 0a 43 41 53 45 28 30 29 0d 0a (){....CASE(0)..
0780: 09 69 6e 74 20 61 6e 73 77 65 72 73 5f 5b 5d 20 .int answers_[]
0790: 3d 20 7b 30 2c 20 31 2c 20 32 2c 20 33 2c 20 34 = {0, 1, 2, 3, 4
07a0: 7d 3b 0d 0a 09 20 20 76 65 63 74 6f 72 20 3c 69 };... vector <i
07b0: 6e 74 3e 20 61 6e 73 77 65 72 73 28 61 6e 73 77 nt> answers(answ
07c0: 65 72 73 5f 2c 20 61 6e 73 77 65 72 73 5f 2b 73 ers_, answers_+s
07d0: 69 7a 65 6f 66 28 61 6e 73 77 65 72 73 5f 29 2f izeof(answers_)/
07e0: 73 69 7a 65 6f 66 28 2a 61 6e 73 77 65 72 73 5f sizeof(*answers_
07f0: 29 29 3b 20 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 )); ...long long
0800: 20 5f 20 3d 20 32 4c 4c 3b 20 0d 0a 45 4e 44 0d _ = 2LL; ..END.
0810: 0a 43 41 53 45 28 31 29 0d 0a 09 69 6e 74 20 61 .CASE(1)...int a
0820: 6e 73 77 65 72 73 5f 5b 5d 20 3d 20 7b 35 2c 20 nswers_[] = {5,
0830: 38 7d 3b 0d 0a 09 20 20 76 65 63 74 6f 72 20 3c 8};... vector <
0840: 69 6e 74 3e 20 61 6e 73 77 65 72 73 28 61 6e 73 int> answers(ans
0850: 77 65 72 73 5f 2c 20 61 6e 73 77 65 72 73 5f 2b wers_, answers_+
0860: 73 69 7a 65 6f 66 28 61 6e 73 77 65 72 73 5f 29 sizeof(answers_)
0870: 2f 73 69 7a 65 6f 66 28 2a 61 6e 73 77 65 72 73 /sizeof(*answers
0880: 5f 29 29 3b 20 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e _)); ...long lon
0890: 67 20 5f 20 3d 20 30 4c 4c 3b 20 0d 0a 45 4e 44 g _ = 0LL; ..END
08a0: 0d 0a 43 41 53 45 28 32 29 0d 0a 09 69 6e 74 20 ..CASE(2)...int
08b0: 61 6e 73 77 65 72 73 5f 5b 5d 20 3d 20 7b 30 2c answers_[] = {0,
08c0: 20 30 2c 20 30 2c 20 30 2c 20 30 2c 20 30 7d 3b 0, 0, 0, 0, 0};
08d0: 0d 0a 09 20 20 76 65 63 74 6f 72 20 3c 69 6e 74 ... vector <int
08e0: 3e 20 61 6e 73 77 65 72 73 28 61 6e 73 77 65 72 > answers(answer
08f0: 73 5f 2c 20 61 6e 73 77 65 72 73 5f 2b 73 69 7a s_, answers_+siz
0900: 65 6f 66 28 61 6e 73 77 65 72 73 5f 29 2f 73 69 eof(answers_)/si
0910: 7a 65 6f 66 28 2a 61 6e 73 77 65 72 73 5f 29 29 zeof(*answers_))
0920: 3b 20 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 20 5f ; ...long long _
0930: 20 3d 20 30 4c 4c 3b 20 0d 0a 45 4e 44 0d 0a 43 = 0LL; ..END..C
0940: 41 53 45 28 33 29 0d 0a 09 69 6e 74 20 61 6e 73 ASE(3)...int ans
0950: 77 65 72 73 5f 5b 5d 20 3d 20 7b 31 2c 20 30 2c wers_[] = {1, 0,
0960: 20 32 2c 20 30 2c 20 31 7d 3b 0d 0a 09 20 20 76 2, 0, 1};... v
0970: 65 63 74 6f 72 20 3c 69 6e 74 3e 20 61 6e 73 77 ector <int> answ
0980: 65 72 73 28 61 6e 73 77 65 72 73 5f 2c 20 61 6e ers(answers_, an
0990: 73 77 65 72 73 5f 2b 73 69 7a 65 6f 66 28 61 6e swers_+sizeof(an
09a0: 73 77 65 72 73 5f 29 2f 73 69 7a 65 6f 66 28 2a swers_)/sizeof(*
09b0: 61 6e 73 77 65 72 73 5f 29 29 3b 20 0d 0a 09 6c answers_)); ...l
09c0: 6f 6e 67 20 6c 6f 6e 67 20 5f 20 3d 20 38 4c 4c ong long _ = 8LL
09d0: 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 34 29 ; ..END..CASE(4)
09e0: 0d 0a 09 69 6e 74 20 61 6e 73 77 65 72 73 5f 5b ...int answers_[
09f0: 5d 20 3d 20 7b 31 2c 20 30 2c 20 31 7d 3b 0d 0a ] = {1, 0, 1};..
0a00: 09 20 20 76 65 63 74 6f 72 20 3c 69 6e 74 3e 20 . vector <int>
0a10: 61 6e 73 77 65 72 73 28 61 6e 73 77 65 72 73 5f answers(answers_
0a20: 2c 20 61 6e 73 77 65 72 73 5f 2b 73 69 7a 65 6f , answers_+sizeo
0a30: 66 28 61 6e 73 77 65 72 73 5f 29 2f 73 69 7a 65 f(answers_)/size
0a40: 6f 66 28 2a 61 6e 73 77 65 72 73 5f 29 29 3b 20 of(*answers_));
0a50: 0d 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 20 5f 20 3d ...long long _ =
0a60: 20 30 4c 4c 3b 20 0d 0a 45 4e 44 0d 0a 2f 2a 0d 0LL; ..END../*.
0a70: 0a 43 41 53 45 28 35 29 0d 0a 09 69 6e 74 20 61 .CASE(5)...int a
0a80: 6e 73 77 65 72 73 5f 5b 5d 20 3d 20 3b 0d 0a 09 nswers_[] = ;...
0a90: 20 20 76 65 63 74 6f 72 20 3c 69 6e 74 3e 20 61 vector <int> a
0aa0: 6e 73 77 65 72 73 28 61 6e 73 77 65 72 73 5f 2c nswers(answers_,
0ab0: 20 61 6e 73 77 65 72 73 5f 2b 73 69 7a 65 6f 66 answers_+sizeof
0ac0: 28 61 6e 73 77 65 72 73 5f 29 2f 73 69 7a 65 6f (answers_)/sizeo
0ad0: 66 28 2a 61 6e 73 77 65 72 73 5f 29 29 3b 20 0d f(*answers_)); .
0ae0: 0a 09 6c 6f 6e 67 20 6c 6f 6e 67 20 5f 20 3d 20 ..long long _ =
0af0: 4c 4c 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 LL; ..END..CASE(
0b00: 36 29 0d 0a 09 69 6e 74 20 61 6e 73 77 65 72 73 6)...int answers
0b10: 5f 5b 5d 20 3d 20 3b 0d 0a 09 20 20 76 65 63 74 _[] = ;... vect
0b20: 6f 72 20 3c 69 6e 74 3e 20 61 6e 73 77 65 72 73 or <int> answers
0b30: 28 61 6e 73 77 65 72 73 5f 2c 20 61 6e 73 77 65 (answers_, answe
0b40: 72 73 5f 2b 73 69 7a 65 6f 66 28 61 6e 73 77 65 rs_+sizeof(answe
0b50: 72 73 5f 29 2f 73 69 7a 65 6f 66 28 2a 61 6e 73 rs_)/sizeof(*ans
0b60: 77 65 72 73 5f 29 29 3b 20 0d 0a 09 6c 6f 6e 67 wers_)); ...long
0b70: 20 6c 6f 6e 67 20 5f 20 3d 20 4c 4c 3b 20 0d 0a long _ = LL; ..
0b80: 45 4e 44 0d 0a 2a 2f 0d 0a 7d 0d 0a 2f 2f 20 45 END..*/..}..// E
0b90: 4e 44 20 43 55 54 20 48 45 52 45 0d 0a ND CUT HERE..