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 75 73 69 6e 67 20 6e 61 6d sert>..using nam
0140: 65 73 70 61 63 65 20 73 74 64 3b 0d 0a 74 79 70 espace std;..typ
0150: 65 64 65 66 20 6c 6f 6e 67 20 6c 6f 6e 67 20 4c edef long long L
0160: 4c 3b 0d 0a 74 79 70 65 64 65 66 20 6c 6f 6e 67 L;..typedef long
0170: 20 64 6f 75 62 6c 65 20 4c 44 3b 0d 0a 74 79 70 double LD;..typ
0180: 65 64 65 66 20 63 6f 6d 70 6c 65 78 3c 4c 44 3e edef complex<LD>
0190: 20 43 4d 50 3b 0d 0a 0d 0a 63 6c 61 73 73 20 53 CMP;....class S
01a0: 61 66 65 52 65 6d 6f 76 61 6c 20 7b 20 70 75 62 afeRemoval { pub
01b0: 6c 69 63 3a 0d 0a 09 69 6e 74 20 72 65 6d 6f 76 lic:...int remov
01c0: 65 54 68 65 6d 28 76 65 63 74 6f 72 20 3c 69 6e eThem(vector <in
01d0: 74 3e 20 73 65 71 2c 20 69 6e 74 20 4b 29 0d 0a t> seq, int K)..
01e0: 09 7b 0d 0a 09 09 73 6f 72 74 28 73 65 71 2e 62 .{....sort(seq.b
01f0: 65 67 69 6e 28 29 2c 20 73 65 71 2e 65 6e 64 28 egin(), seq.end(
0200: 29 29 3b 0d 0a 09 09 63 6f 6e 73 74 20 69 6e 74 ));....const int
0210: 20 73 75 6d 20 3d 20 61 63 63 75 6d 75 6c 61 74 sum = accumulat
0220: 65 28 73 65 71 2e 62 65 67 69 6e 28 29 2c 20 73 e(seq.begin(), s
0230: 65 71 2e 65 6e 64 28 29 2c 20 30 29 3b 0d 0a 0d eq.end(), 0);...
0240: 0a 09 09 76 65 63 74 6f 72 3c 69 6e 74 3e 20 63 ...vector<int> c
0250: 6c 73 5b 34 5d 3b 0d 0a 09 09 66 6f 72 28 69 6e ls[4];....for(in
0260: 74 20 69 3d 30 3b 20 69 3c 73 65 71 2e 73 69 7a t i=0; i<seq.siz
0270: 65 28 29 3b 20 2b 2b 69 29 0d 0a 09 09 09 63 6c e(); ++i).....cl
0280: 73 5b 73 65 71 5b 69 5d 25 34 5d 2e 70 75 73 68 s[seq[i]%4].push
0290: 5f 62 61 63 6b 28 73 65 71 5b 69 5d 29 3b 0d 0a _back(seq[i]);..
02a0: 0d 0a 09 09 69 6e 74 20 62 65 73 74 20 3d 20 2d ....int best = -
02b0: 31 3b 0d 0a 09 09 66 6f 72 28 69 6e 74 20 72 3d 1;....for(int r=
02c0: 30 3b 20 72 3c 34 3b 20 2b 2b 72 29 20 7b 0d 0a 0; r<4; ++r) {..
02d0: 09 09 09 69 6e 74 20 62 20 3d 20 72 65 63 28 63 ...int b = rec(c
02e0: 6c 73 2c 20 30 2c 20 30 2c 20 30 2c 20 30 2c 20 ls, 0, 0, 0, 0,
02f0: 4b 2c 20 73 75 6d 25 34 29 3b 0d 0a 09 09 09 69 K, sum%4);.....i
0300: 66 28 62 20 21 3d 20 2d 31 29 0d 0a 09 09 09 09 f(b != -1)......
0310: 62 65 73 74 20 3d 20 6d 61 78 28 62 65 73 74 2c best = max(best,
0320: 20 73 75 6d 2d 62 29 3b 0d 0a 09 09 7d 0d 0a 09 sum-b);....}...
0330: 09 72 65 74 75 72 6e 20 62 65 73 74 3b 0d 0a 09 .return best;...
0340: 7d 0d 0a 0d 0a 09 6d 61 70 3c 69 6e 74 2c 20 69 }.....map<int, i
0350: 6e 74 3e 20 6d 65 6d 6f 3b 0d 0a 09 69 6e 74 20 nt> memo;...int
0360: 72 65 63 28 63 6f 6e 73 74 20 76 65 63 74 6f 72 rec(const vector
0370: 3c 69 6e 74 3e 20 63 6c 73 5b 34 5d 2c 20 69 6e <int> cls[4], in
0380: 74 20 61 2c 20 69 6e 74 20 62 2c 20 69 6e 74 20 t a, int b, int
0390: 63 2c 20 69 6e 74 20 64 2c 20 69 6e 74 20 4b 2c c, int d, int K,
03a0: 20 69 6e 74 20 52 29 0d 0a 09 7b 0d 0a 09 09 69 int R)...{....i
03b0: 66 28 4b 3d 3d 30 29 0d 0a 09 09 09 72 65 74 75 f(K==0).....retu
03c0: 72 6e 20 28 52 3d 3d 30 20 3f 20 2d 31 20 3a 20 rn (R==0 ? -1 :
03d0: 30 29 3b 0d 0a 09 09 63 6f 6e 73 74 20 69 6e 74 0);....const int
03e0: 20 6b 65 79 20 3d 20 28 28 28 28 4b 2a 35 31 2b key = ((((K*51+
03f0: 61 29 2a 35 31 2b 62 29 2a 35 31 2b 63 29 2a 35 a)*51+b)*51+c)*5
0400: 31 2b 64 29 2a 34 2b 52 3b 0d 0a 09 09 69 66 28 1+d)*4+R;....if(
0410: 6d 65 6d 6f 2e 63 6f 75 6e 74 28 6b 65 79 29 29 memo.count(key))
0420: 0d 0a 09 09 09 72 65 74 75 72 6e 20 6d 65 6d 6f .....return memo
0430: 5b 6b 65 79 5d 3b 0d 0a 0d 0a 09 09 69 6e 74 20 [key];......int
0440: 69 5b 34 5d 20 3d 20 7b 61 2c 62 2c 63 2c 64 7d i[4] = {a,b,c,d}
0450: 3b 0d 0a 09 09 69 6e 74 20 62 65 73 74 20 3d 20 ;....int best =
0460: 2d 31 3b 0d 0a 09 09 66 6f 72 28 69 6e 74 20 6b -1;....for(int k
0470: 3d 30 3b 20 6b 3c 34 3b 20 2b 2b 6b 29 20 69 66 =0; k<4; ++k) if
0480: 28 52 21 3d 6b 20 26 26 20 69 5b 6b 5d 3c 63 6c (R!=k && i[k]<cl
0490: 73 5b 6b 5d 2e 73 69 7a 65 28 29 29 20 7b 0d 0a s[k].size()) {..
04a0: 09 09 09 69 5b 6b 5d 2b 2b 3b 0d 0a 09 09 09 69 ...i[k]++;.....i
04b0: 6e 74 20 62 20 3d 20 72 65 63 28 63 6c 73 2c 20 nt b = rec(cls,
04c0: 69 5b 30 5d 2c 20 69 5b 31 5d 2c 20 69 5b 32 5d i[0], i[1], i[2]
04d0: 2c 20 69 5b 33 5d 2c 20 4b 2d 31 2c 20 28 52 2d , i[3], K-1, (R-
04e0: 6b 2b 34 29 25 34 29 3b 0d 0a 09 09 09 69 5b 6b k+4)%4);.....i[k
04f0: 5d 2d 2d 3b 0d 0a 09 09 09 69 66 28 62 20 21 3d ]--;.....if(b !=
0500: 20 2d 31 29 20 7b 0d 0a 09 09 09 09 62 20 2b 3d -1) {......b +=
0510: 20 63 6c 73 5b 6b 5d 5b 69 5b 6b 5d 5d 3b 0d 0a cls[k][i[k]];..
0520: 09 09 09 09 62 65 73 74 20 3d 20 28 62 65 73 74 ....best = (best
0530: 3d 3d 2d 31 20 3f 20 62 20 3a 20 6d 69 6e 28 62 ==-1 ? b : min(b
0540: 65 73 74 2c 20 62 29 29 3b 0d 0a 09 09 09 7d 0d est, b));.....}.
0550: 0a 09 09 7d 0d 0a 09 09 72 65 74 75 72 6e 20 6d ...}....return m
0560: 65 6d 6f 5b 6b 65 79 5d 20 3d 20 62 65 73 74 3b emo[key] = best;
0570: 0d 0a 09 7d 0d 0a 7d 3b 0d 0a 0d 0a 2f 2f 20 42 ...}..};....// B
0580: 45 47 49 4e 20 43 55 54 20 48 45 52 45 0d 0a 23 EGIN CUT HERE..#
0590: 69 6e 63 6c 75 64 65 20 3c 63 74 69 6d 65 3e 0d include <ctime>.
05a0: 0a 64 6f 75 62 6c 65 20 73 74 61 72 74 5f 74 69 .double start_ti
05b0: 6d 65 3b 20 73 74 72 69 6e 67 20 74 69 6d 65 72 me; string timer
05c0: 28 29 0d 0a 20 7b 20 6f 73 74 72 69 6e 67 73 74 ().. { ostringst
05d0: 72 65 61 6d 20 6f 73 3b 20 6f 73 20 3c 3c 20 22 ream os; os << "
05e0: 20 28 22 20 3c 3c 20 69 6e 74 28 28 63 6c 6f 63 (" << int((cloc
05f0: 6b 28 29 2d 73 74 61 72 74 5f 74 69 6d 65 29 2f k()-start_time)/
0600: 43 4c 4f 43 4b 53 5f 50 45 52 5f 53 45 43 2a 31 CLOCKS_PER_SEC*1
0610: 30 30 30 29 20 3c 3c 20 22 20 6d 73 65 63 29 22 000) << " msec)"
0620: 3b 20 72 65 74 75 72 6e 20 6f 73 2e 73 74 72 28 ; return os.str(
0630: 29 3b 20 7d 0d 0a 74 65 6d 70 6c 61 74 65 3c 74 ); }..template<t
0640: 79 70 65 6e 61 6d 65 20 54 3e 20 6f 73 74 72 65 ypename T> ostre
0650: 61 6d 26 20 6f 70 65 72 61 74 6f 72 3c 3c 28 6f am& operator<<(o
0660: 73 74 72 65 61 6d 26 20 6f 73 2c 20 63 6f 6e 73 stream& os, cons
0670: 74 20 76 65 63 74 6f 72 3c 54 3e 26 20 76 29 0d t vector<T>& v).
0680: 0a 20 7b 20 6f 73 20 3c 3c 20 22 7b 20 22 3b 0d . { os << "{ ";.
0690: 0a 20 20 20 66 6f 72 28 74 79 70 65 6e 61 6d 65 . for(typename
06a0: 20 76 65 63 74 6f 72 3c 54 3e 3a 3a 63 6f 6e 73 vector<T>::cons
06b0: 74 5f 69 74 65 72 61 74 6f 72 20 69 74 3d 76 2e t_iterator it=v.
06c0: 62 65 67 69 6e 28 29 3b 20 69 74 21 3d 76 2e 65 begin(); it!=v.e
06d0: 6e 64 28 29 3b 20 2b 2b 69 74 29 0d 0a 20 20 20 nd(); ++it)..
06e0: 6f 73 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 2a 69 os << '\"' << *i
06f0: 74 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 28 69 74 t << '\"' << (it
0700: 2b 31 3d 3d 76 2e 65 6e 64 28 29 20 3f 20 22 22 +1==v.end() ? ""
0710: 20 3a 20 22 2c 20 22 29 3b 20 6f 73 20 3c 3c 20 : ", "); os <<
0720: 22 20 7d 22 3b 20 72 65 74 75 72 6e 20 6f 73 3b " }"; return os;
0730: 20 7d 0d 0a 76 6f 69 64 20 76 65 72 69 66 79 5f }..void verify_
0740: 63 61 73 65 28 63 6f 6e 73 74 20 69 6e 74 26 20 case(const int&
0750: 45 78 70 65 63 74 65 64 2c 20 63 6f 6e 73 74 20 Expected, const
0760: 69 6e 74 26 20 52 65 63 65 69 76 65 64 29 20 7b int& Received) {
0770: 0d 0a 20 62 6f 6f 6c 20 6f 6b 20 3d 20 28 45 78 .. bool ok = (Ex
0780: 70 65 63 74 65 64 20 3d 3d 20 52 65 63 65 69 76 pected == Receiv
0790: 65 64 29 3b 0d 0a 20 69 66 28 6f 6b 29 20 63 65 ed);.. if(ok) ce
07a0: 72 72 20 3c 3c 20 22 50 41 53 53 45 44 22 20 3c rr << "PASSED" <
07b0: 3c 20 74 69 6d 65 72 28 29 20 3c 3c 20 65 6e 64 < timer() << end
07c0: 6c 3b 20 20 65 6c 73 65 20 7b 20 63 65 72 72 20 l; else { cerr
07d0: 3c 3c 20 22 46 41 49 4c 45 44 22 20 3c 3c 20 74 << "FAILED" << t
07e0: 69 6d 65 72 28 29 20 3c 3c 20 65 6e 64 6c 3b 0d imer() << endl;.
07f0: 0a 20 63 65 72 72 20 3c 3c 20 22 5c 74 6f 3a 20 . cerr << "\to:
0800: 5c 22 22 20 3c 3c 20 45 78 70 65 63 74 65 64 20 \"" << Expected
0810: 3c 3c 20 27 5c 22 27 20 3c 3c 20 65 6e 64 6c 20 << '\"' << endl
0820: 3c 3c 20 22 5c 74 78 3a 20 5c 22 22 20 3c 3c 20 << "\tx: \"" <<
0830: 52 65 63 65 69 76 65 64 20 3c 3c 20 27 5c 22 27 Received << '\"'
0840: 20 3c 3c 20 65 6e 64 6c 3b 20 7d 20 7d 0d 0a 23 << endl; } }..#
0850: 64 65 66 69 6e 65 20 43 41 53 45 28 4e 29 20 7b define CASE(N) {
0860: 63 65 72 72 20 3c 3c 20 22 54 65 73 74 20 43 61 cerr << "Test Ca
0870: 73 65 20 23 22 20 3c 3c 20 4e 20 3c 3c 20 22 2e se #" << N << ".
0880: 2e 2e 22 20 3c 3c 20 66 6c 75 73 68 3b 20 73 74 .." << flush; st
0890: 61 72 74 5f 74 69 6d 65 3d 63 6c 6f 63 6b 28 29 art_time=clock()
08a0: 3b 0d 0a 23 64 65 66 69 6e 65 20 45 4e 44 09 20 ;..#define END.
08b0: 76 65 72 69 66 79 5f 63 61 73 65 28 5f 2c 20 53 verify_case(_, S
08c0: 61 66 65 52 65 6d 6f 76 61 6c 28 29 2e 72 65 6d afeRemoval().rem
08d0: 6f 76 65 54 68 65 6d 28 73 65 71 2c 20 6b 29 29 oveThem(seq, k))
08e0: 3b 7d 0d 0a 69 6e 74 20 6d 61 69 6e 28 29 7b 0d ;}..int main(){.
08f0: 0a 0d 0a 43 41 53 45 28 30 29 0d 0a 09 69 6e 74 ...CASE(0)...int
0900: 20 73 65 71 5f 5b 5d 20 3d 20 7b 33 2c 20 38 2c seq_[] = {3, 8,
0910: 20 34 7d 3b 0d 0a 09 20 20 76 65 63 74 6f 72 20 4};... vector
0920: 3c 69 6e 74 3e 20 73 65 71 28 73 65 71 5f 2c 20 <int> seq(seq_,
0930: 73 65 71 5f 2b 73 69 7a 65 6f 66 28 73 65 71 5f seq_+sizeof(seq_
0940: 29 2f 73 69 7a 65 6f 66 28 2a 73 65 71 5f 29 29 )/sizeof(*seq_))
0950: 3b 20 0d 0a 09 69 6e 74 20 6b 20 3d 20 31 3b 20 ; ...int k = 1;
0960: 0d 0a 09 69 6e 74 20 5f 20 3d 20 31 31 3b 20 0d ...int _ = 11; .
0970: 0a 45 4e 44 0d 0a 43 41 53 45 28 31 29 0d 0a 09 .END..CASE(1)...
0980: 69 6e 74 20 73 65 71 5f 5b 5d 20 3d 20 7b 31 2c int seq_[] = {1,
0990: 31 2c 31 2c 31 2c 31 2c 31 7d 3b 0d 0a 09 20 20 1,1,1,1,1};...
09a0: 76 65 63 74 6f 72 20 3c 69 6e 74 3e 20 73 65 71 vector <int> seq
09b0: 28 73 65 71 5f 2c 20 73 65 71 5f 2b 73 69 7a 65 (seq_, seq_+size
09c0: 6f 66 28 73 65 71 5f 29 2f 73 69 7a 65 6f 66 28 of(seq_)/sizeof(
09d0: 2a 73 65 71 5f 29 29 3b 20 0d 0a 09 69 6e 74 20 *seq_)); ...int
09e0: 6b 20 3d 20 33 3b 20 0d 0a 09 69 6e 74 20 5f 20 k = 3; ...int _
09f0: 3d 20 2d 31 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 = -1; ..END..CAS
0a00: 45 28 32 29 0d 0a 09 69 6e 74 20 73 65 71 5f 5b E(2)...int seq_[
0a10: 5d 20 3d 20 7b 31 2c 36 2c 31 2c 31 30 2c 31 2c ] = {1,6,1,10,1,
0a20: 32 2c 37 2c 31 31 7d 3b 0d 0a 09 20 20 76 65 63 2,7,11};... vec
0a30: 74 6f 72 20 3c 69 6e 74 3e 20 73 65 71 28 73 65 tor <int> seq(se
0a40: 71 5f 2c 20 73 65 71 5f 2b 73 69 7a 65 6f 66 28 q_, seq_+sizeof(
0a50: 73 65 71 5f 29 2f 73 69 7a 65 6f 66 28 2a 73 65 seq_)/sizeof(*se
0a60: 71 5f 29 29 3b 20 0d 0a 09 69 6e 74 20 6b 20 3d q_)); ...int k =
0a70: 20 36 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 32 6; ...int _ = 2
0a80: 31 3b 20 0d 0a 45 4e 44 0d 0a 43 41 53 45 28 33 1; ..END..CASE(3
0a90: 29 0d 0a 09 69 6e 74 20 73 65 71 5f 5b 5d 20 3d )...int seq_[] =
0aa0: 20 7b 31 2c 31 2c 31 2c 31 2c 31 2c 31 2c 31 2c {1,1,1,1,1,1,1,
0ab0: 31 2c 37 7d 3b 0d 0a 09 20 20 76 65 63 74 6f 72 1,7};... vector
0ac0: 20 3c 69 6e 74 3e 20 73 65 71 28 73 65 71 5f 2c <int> seq(seq_,
0ad0: 20 73 65 71 5f 2b 73 69 7a 65 6f 66 28 73 65 71 seq_+sizeof(seq
0ae0: 5f 29 2f 73 69 7a 65 6f 66 28 2a 73 65 71 5f 29 _)/sizeof(*seq_)
0af0: 29 3b 20 0d 0a 09 69 6e 74 20 6b 20 3d 20 33 3b ); ...int k = 3;
0b00: 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 36 3b 20 0d ...int _ = 6; .
0b10: 0a 45 4e 44 0d 0a 2f 2a 0d 0a 43 41 53 45 28 34 .END../*..CASE(4
0b20: 29 0d 0a 09 69 6e 74 20 73 65 71 5f 5b 5d 20 3d )...int seq_[] =
0b30: 20 3b 0d 0a 09 20 20 76 65 63 74 6f 72 20 3c 69 ;... vector <i
0b40: 6e 74 3e 20 73 65 71 28 73 65 71 5f 2c 20 73 65 nt> seq(seq_, se
0b50: 71 5f 2b 73 69 7a 65 6f 66 28 73 65 71 5f 29 2f q_+sizeof(seq_)/
0b60: 73 69 7a 65 6f 66 28 2a 73 65 71 5f 29 29 3b 20 sizeof(*seq_));
0b70: 0d 0a 09 69 6e 74 20 6b 20 3d 20 3b 20 0d 0a 09 ...int k = ; ...
0b80: 69 6e 74 20 5f 20 3d 20 3b 20 0d 0a 45 4e 44 0d int _ = ; ..END.
0b90: 0a 43 41 53 45 28 35 29 0d 0a 09 69 6e 74 20 73 .CASE(5)...int s
0ba0: 65 71 5f 5b 5d 20 3d 20 3b 0d 0a 09 20 20 76 65 eq_[] = ;... ve
0bb0: 63 74 6f 72 20 3c 69 6e 74 3e 20 73 65 71 28 73 ctor <int> seq(s
0bc0: 65 71 5f 2c 20 73 65 71 5f 2b 73 69 7a 65 6f 66 eq_, seq_+sizeof
0bd0: 28 73 65 71 5f 29 2f 73 69 7a 65 6f 66 28 2a 73 (seq_)/sizeof(*s
0be0: 65 71 5f 29 29 3b 20 0d 0a 09 69 6e 74 20 6b 20 eq_)); ...int k
0bf0: 3d 20 3b 20 0d 0a 09 69 6e 74 20 5f 20 3d 20 3b = ; ...int _ = ;
0c00: 20 0d 0a 45 4e 44 0d 0a 2a 2f 0d 0a 7d 0d 0a 2f ..END..*/..}../
0c10: 2f 20 45 4e 44 20 43 55 54 20 48 45 52 45 0d 0a / END CUT HERE..