0000: 23 69 6e 63 6c 75 64 65 20 3c 69 6f 73 74 72 65 #include <iostre
0010: 61 6d 3e 0a 23 69 6e 63 6c 75 64 65 20 3c 73 73 am>.#include <ss
0020: 74 72 65 61 6d 3e 0a 23 69 6e 63 6c 75 64 65 20 tream>.#include
0030: 3c 69 6f 6d 61 6e 69 70 3e 0a 23 69 6e 63 6c 75 <iomanip>.#inclu
0040: 64 65 20 3c 76 65 63 74 6f 72 3e 0a 23 69 6e 63 de <vector>.#inc
0050: 6c 75 64 65 20 3c 73 74 72 69 6e 67 3e 0a 23 69 lude <string>.#i
0060: 6e 63 6c 75 64 65 20 3c 6d 61 70 3e 0a 23 69 6e nclude <map>.#in
0070: 63 6c 75 64 65 20 3c 73 65 74 3e 0a 23 69 6e 63 clude <set>.#inc
0080: 6c 75 64 65 20 3c 61 6c 67 6f 72 69 74 68 6d 3e lude <algorithm>
0090: 0a 23 69 6e 63 6c 75 64 65 20 3c 6e 75 6d 65 72 .#include <numer
00a0: 69 63 3e 0a 23 69 6e 63 6c 75 64 65 20 3c 69 74 ic>.#include <it
00b0: 65 72 61 74 6f 72 3e 0a 23 69 6e 63 6c 75 64 65 erator>.#include
00c0: 20 3c 6d 65 6d 6f 72 79 3e 0a 23 69 6e 63 6c 75 <memory>.#inclu
00d0: 64 65 20 3c 66 75 6e 63 74 69 6f 6e 61 6c 3e 0a de <functional>.
00e0: 23 69 6e 63 6c 75 64 65 20 3c 63 6f 6d 70 6c 65 #include <comple
00f0: 78 3e 0a 23 69 6e 63 6c 75 64 65 20 3c 71 75 65 x>.#include <que
0100: 75 65 3e 0a 23 69 6e 63 6c 75 64 65 20 3c 73 74 ue>.#include <st
0110: 61 63 6b 3e 0a 23 69 6e 63 6c 75 64 65 20 3c 63 ack>.#include <c
0120: 6d 61 74 68 3e 0a 23 69 6e 63 6c 75 64 65 20 3c math>.#include <
0130: 63 61 73 73 65 72 74 3e 0a 75 73 69 6e 67 20 6e cassert>.using n
0140: 61 6d 65 73 70 61 63 65 20 73 74 64 3b 0a 74 79 amespace std;.ty
0150: 70 65 64 65 66 20 6c 6f 6e 67 20 6c 6f 6e 67 20 pedef long long
0160: 4c 4c 3b 0a 74 79 70 65 64 65 66 20 6c 6f 6e 67 LL;.typedef long
0170: 20 64 6f 75 62 6c 65 20 4c 44 3b 0a 74 79 70 65 double LD;.type
0180: 64 65 66 20 63 6f 6d 70 6c 65 78 3c 4c 44 3e 20 def complex<LD>
0190: 43 4d 50 3b 0a 0a 74 65 6d 70 6c 61 74 65 3c 74 CMP;..template<t
01a0: 79 70 65 6e 61 6d 65 20 54 3e 0a 63 6c 61 73 73 ypename T>.class
01b0: 20 49 64 47 65 6e 0a 7b 0a 09 6d 61 70 3c 54 2c IdGen.{..map<T,
01c0: 20 69 6e 74 3e 20 76 32 69 64 5f 3b 0a 09 76 65 int> v2id_;..ve
01d0: 63 74 6f 72 3c 54 3e 20 20 20 69 64 32 76 5f 3b ctor<T> id2v_;
01e0: 0a 70 75 62 6c 69 63 3a 0a 09 69 6e 74 20 76 32 .public:..int v2
01f0: 69 64 28 63 6f 6e 73 74 20 54 26 20 76 29 20 7b id(const T& v) {
0200: 0a 09 09 69 66 28 20 21 76 32 69 64 5f 2e 63 6f ...if( !v2id_.co
0210: 75 6e 74 28 76 29 20 29 20 7b 20 76 32 69 64 5f unt(v) ) { v2id_
0220: 5b 76 5d 20 3d 20 73 69 7a 65 28 29 3b 20 69 64 [v] = size(); id
0230: 32 76 5f 2e 70 75 73 68 5f 62 61 63 6b 28 76 29 2v_.push_back(v)
0240: 3b 20 7d 0a 09 09 72 65 74 75 72 6e 20 76 32 69 ; }...return v2i
0250: 64 5f 5b 76 5d 3b 0a 09 7d 0a 09 63 6f 6e 73 74 d_[v];..}..const
0260: 20 54 26 20 69 64 32 76 28 69 6e 74 20 69 29 20 T& id2v(int i)
0270: 63 6f 6e 73 74 20 7b 20 72 65 74 75 72 6e 20 69 const { return i
0280: 64 32 76 5f 5b 69 5d 3b 20 7d 0a 09 69 6e 74 20 d2v_[i]; }..int
0290: 73 69 7a 65 28 29 20 63 6f 6e 73 74 20 7b 20 72 size() const { r
02a0: 65 74 75 72 6e 20 69 64 32 76 5f 2e 73 69 7a 65 eturn id2v_.size
02b0: 28 29 3b 20 7d 0a 7d 3b 0a 0a 74 65 6d 70 6c 61 (); }.};..templa
02c0: 74 65 3c 74 79 70 65 6e 61 6d 65 20 56 65 72 74 te<typename Vert
02d0: 2c 20 74 79 70 65 6e 61 6d 65 20 46 6c 6f 77 2c , typename Flow,
02e0: 20 69 6e 74 20 4e 56 3d 35 31 32 3e 0a 63 6c 61 int NV=512>.cla
02f0: 73 73 20 4d 61 78 46 6c 6f 77 0a 7b 0a 09 49 64 ss MaxFlow.{..Id
0300: 47 65 6e 3c 56 65 72 74 3e 20 69 64 67 65 6e 3b Gen<Vert> idgen;
0310: 0a 09 76 65 63 74 6f 72 3c 69 6e 74 3e 20 47 5b ..vector<int> G[
0320: 4e 56 5d 3b 0a 09 46 6c 6f 77 20 46 5b 4e 56 5d NV];..Flow F[NV]
0330: 5b 4e 56 5d 3b 0a 0a 70 75 62 6c 69 63 3a 0a 09 [NV];..public:..
0340: 76 6f 69 64 20 61 64 64 45 64 67 65 28 20 56 65 void addEdge( Ve
0350: 72 74 20 73 5f 2c 20 56 65 72 74 20 74 5f 2c 20 rt s_, Vert t_,
0360: 46 6c 6f 77 20 66 20 29 0a 09 7b 0a 09 09 63 6f Flow f )..{...co
0370: 6e 73 74 20 69 6e 74 20 73 20 3d 20 69 64 67 65 nst int s = idge
0380: 6e 2e 76 32 69 64 28 73 5f 29 2c 20 74 20 3d 20 n.v2id(s_), t =
0390: 69 64 67 65 6e 2e 76 32 69 64 28 74 5f 29 3b 0a idgen.v2id(t_);.
03a0: 09 09 47 5b 73 5d 2e 70 75 73 68 5f 62 61 63 6b ..G[s].push_back
03b0: 28 74 29 3b 0a 09 09 47 5b 74 5d 2e 70 75 73 68 (t);...G[t].push
03c0: 5f 62 61 63 6b 28 73 29 3b 0a 09 09 46 5b 73 5d _back(s);...F[s]
03d0: 5b 74 5d 20 3d 20 66 3b 0a 09 09 46 5b 74 5d 5b [t] = f;...F[t][
03e0: 73 5d 20 3d 20 30 3b 0a 09 7d 0a 0a 09 46 6c 6f s] = 0;..}...Flo
03f0: 77 20 63 61 6c 63 28 20 56 65 72 74 20 73 5f 2c w calc( Vert s_,
0400: 20 56 65 72 74 20 74 5f 20 29 0a 09 7b 0a 09 09 Vert t_ )..{...
0410: 63 6f 6e 73 74 20 69 6e 74 20 53 20 3d 20 69 64 const int S = id
0420: 67 65 6e 2e 76 32 69 64 28 73 5f 29 2c 20 44 20 gen.v2id(s_), D
0430: 3d 20 69 64 67 65 6e 2e 76 32 69 64 28 74 5f 29 = idgen.v2id(t_)
0440: 3b 0a 09 09 66 6f 72 28 20 46 6c 6f 77 20 74 6f ;...for( Flow to
0450: 74 61 6c 3d 30 20 3b 3b 20 29 20 7b 0a 09 09 09 tal=0 ;; ) {....
0460: 2f 2f 20 44 6f 20 42 46 53 20 61 6e 64 20 63 6f // Do BFS and co
0470: 6d 70 75 74 65 20 74 68 65 20 6c 65 76 65 6c 20 mpute the level
0480: 66 6f 72 20 65 61 63 68 20 6e 6f 64 65 2e 0a 09 for each node...
0490: 09 09 69 6e 74 20 4c 56 5b 4e 56 5d 20 3d 20 7b ..int LV[NV] = {
04a0: 30 7d 3b 0a 09 09 09 76 65 63 74 6f 72 3c 69 6e 0};....vector<in
04b0: 74 3e 20 51 28 31 2c 20 53 29 3b 0a 09 09 09 66 t> Q(1, S);....f
04c0: 6f 72 28 69 6e 74 20 6c 76 3d 31 3b 20 21 51 2e or(int lv=1; !Q.
04d0: 65 6d 70 74 79 28 29 3b 20 2b 2b 6c 76 29 20 7b empty(); ++lv) {
04e0: 0a 09 09 09 09 76 65 63 74 6f 72 3c 69 6e 74 3e .....vector<int>
04f0: 20 51 32 3b 0a 09 09 09 09 66 6f 72 28 73 69 7a Q2;.....for(siz
0500: 65 5f 74 20 69 3d 30 3b 20 69 21 3d 51 2e 73 69 e_t i=0; i!=Q.si
0510: 7a 65 28 29 3b 20 2b 2b 69 29 20 7b 0a 09 09 09 ze(); ++i) {....
0520: 09 09 63 6f 6e 73 74 20 76 65 63 74 6f 72 3c 69 ..const vector<i
0530: 6e 74 3e 26 20 6e 65 20 3d 20 47 5b 51 5b 69 5d nt>& ne = G[Q[i]
0540: 5d 3b 0a 09 09 09 09 09 66 6f 72 28 73 69 7a 65 ];......for(size
0550: 5f 74 20 6a 3d 30 3b 20 6a 21 3d 6e 65 2e 73 69 _t j=0; j!=ne.si
0560: 7a 65 28 29 3b 20 2b 2b 6a 29 0a 09 09 09 09 09 ze(); ++j)......
0570: 09 69 66 28 20 46 5b 51 5b 69 5d 5d 5b 6e 65 5b .if( F[Q[i]][ne[
0580: 6a 5d 5d 20 26 26 20 21 4c 56 5b 6e 65 5b 6a 5d j]] && !LV[ne[j]
0590: 5d 20 26 26 20 6e 65 5b 6a 5d 21 3d 53 20 29 0a ] && ne[j]!=S ).
05a0: 09 09 09 09 09 09 09 4c 56 5b 6e 65 5b 6a 5d 5d .......LV[ne[j]]
05b0: 3d 6c 76 2c 20 51 32 2e 70 75 73 68 5f 62 61 63 =lv, Q2.push_bac
05c0: 6b 28 6e 65 5b 6a 5d 29 3b 0a 09 09 09 09 7d 0a k(ne[j]);.....}.
05d0: 09 09 09 09 51 2e 73 77 61 70 28 51 32 29 3b 0a ....Q.swap(Q2);.
05e0: 09 09 09 7d 0a 0a 09 09 09 2f 2f 20 44 65 73 74 ...}.....// Dest
05f0: 69 6e 61 74 69 6f 6e 20 69 73 20 6e 6f 77 20 75 ination is now u
0600: 6e 72 65 61 63 68 61 62 6c 65 2e 20 44 6f 6e 65 nreachable. Done
0610: 2e 0a 09 09 09 69 66 28 20 21 4c 56 5b 44 5d 20 .....if( !LV[D]
0620: 29 0a 09 09 09 09 72 65 74 75 72 6e 20 74 6f 74 ).....return tot
0630: 61 6c 3b 0a 0a 09 09 09 2f 2f 20 49 74 65 72 61 al;.....// Itera
0640: 74 69 6e 67 20 44 46 53 2e 0a 09 09 09 62 6f 6f ting DFS.....boo
0650: 6c 20 62 6c 6f 63 6b 65 64 5b 4e 56 5d 20 3d 20 l blocked[NV] =
0660: 7b 7d 3b 0a 09 09 09 74 6f 74 61 6c 20 2b 3d 20 {};....total +=
0670: 64 69 6e 69 63 5f 64 66 73 28 20 53 2c 20 44 2c dinic_dfs( S, D,
0680: 20 4c 56 2c 20 30 78 37 66 66 66 66 66 66 66 2c LV, 0x7fffffff,
0690: 20 62 6c 6f 63 6b 65 64 20 29 3b 0a 09 09 7d 0a blocked );...}.
06a0: 09 7d 0a 0a 70 72 69 76 61 74 65 3a 0a 09 46 6c .}..private:..Fl
06b0: 6f 77 20 64 69 6e 69 63 5f 64 66 73 28 20 69 6e ow dinic_dfs( in
06c0: 74 20 76 2c 20 69 6e 74 20 44 2c 20 69 6e 74 20 t v, int D, int
06d0: 4c 56 5b 5d 2c 20 46 6c 6f 77 20 66 6c 6f 77 5f LV[], Flow flow_
06e0: 69 6e 2c 20 62 6f 6f 6c 20 62 6c 6f 63 6b 65 64 in, bool blocked
06f0: 5b 5d 20 29 0a 09 7b 0a 09 09 46 6c 6f 77 20 66 [] )..{...Flow f
0700: 6c 6f 77 5f 6f 75 74 20 3d 20 30 3b 0a 09 09 66 low_out = 0;...f
0710: 6f 72 28 73 69 7a 65 5f 74 20 69 3d 30 3b 20 69 or(size_t i=0; i
0720: 21 3d 47 5b 76 5d 2e 73 69 7a 65 28 29 3b 20 2b !=G[v].size(); +
0730: 2b 69 29 20 7b 0a 09 09 09 69 6e 74 20 75 20 3d +i) {....int u =
0740: 20 47 5b 76 5d 5b 69 5d 3b 0a 09 09 09 69 66 28 G[v][i];....if(
0750: 20 4c 56 5b 76 5d 2b 31 3d 3d 4c 56 5b 75 5d 20 LV[v]+1==LV[u]
0760: 26 26 20 46 5b 76 5d 5b 75 5d 20 29 20 7b 0a 09 && F[v][u] ) {..
0770: 09 09 09 46 6c 6f 77 20 66 20 3d 20 6d 69 6e 28 ...Flow f = min(
0780: 66 6c 6f 77 5f 69 6e 2d 66 6c 6f 77 5f 6f 75 74 flow_in-flow_out
0790: 2c 20 46 5b 76 5d 5b 75 5d 29 3b 0a 09 09 09 09 , F[v][u]);.....
07a0: 69 66 28 20 75 3d 3d 44 20 7c 7c 20 21 62 6c 6f if( u==D || !blo
07b0: 63 6b 65 64 5b 75 5d 20 26 26 20 28 66 3d 64 69 cked[u] && (f=di
07c0: 6e 69 63 5f 64 66 73 28 75 2c 44 2c 4c 56 2c 66 nic_dfs(u,D,LV,f
07d0: 2c 62 6c 6f 63 6b 65 64 29 29 3e 30 20 29 20 7b ,blocked))>0 ) {
07e0: 0a 09 09 09 09 09 46 5b 76 5d 5b 75 5d 20 20 2d ......F[v][u] -
07f0: 3d 20 66 3b 0a 09 09 09 09 09 46 5b 75 5d 5b 76 = f;......F[u][v
0800: 5d 20 20 2b 3d 20 66 3b 0a 09 09 09 09 09 66 6c ] += f;......fl
0810: 6f 77 5f 6f 75 74 20 2b 3d 20 66 3b 0a 09 09 09 ow_out += f;....
0820: 09 09 69 66 28 20 66 6c 6f 77 5f 69 6e 20 3d 3d ..if( flow_in ==
0830: 20 66 6c 6f 77 5f 6f 75 74 20 29 20 72 65 74 75 flow_out ) retu
0840: 72 6e 20 66 6c 6f 77 5f 6f 75 74 3b 0a 09 09 09 rn flow_out;....
0850: 09 7d 0a 09 09 09 7d 0a 09 09 7d 0a 09 09 62 6c .}....}...}...bl
0860: 6f 63 6b 65 64 5b 76 5d 20 3d 20 28 66 6c 6f 77 ocked[v] = (flow
0870: 5f 6f 75 74 3d 3d 30 29 3b 0a 09 09 72 65 74 75 _out==0);...retu
0880: 72 6e 20 66 6c 6f 77 5f 6f 75 74 3b 0a 09 7d 0a rn flow_out;..}.
0890: 7d 3b 0a 0a 63 6c 61 73 73 20 44 61 6e 63 69 6e };..class Dancin
08a0: 67 50 61 72 74 79 20 7b 20 70 75 62 6c 69 63 3a gParty { public:
08b0: 0a 09 69 6e 74 20 6d 61 78 44 61 6e 63 65 73 28 ..int maxDances(
08c0: 76 65 63 74 6f 72 20 3c 73 74 72 69 6e 67 3e 20 vector <string>
08d0: 6c 69 6b 65 73 2c 20 69 6e 74 20 6b 29 0a 09 7b likes, int k)..{
08e0: 0a 09 09 63 6f 6e 73 74 20 69 6e 74 20 6e 20 3d ...const int n =
08f0: 20 6c 69 6b 65 73 2e 73 69 7a 65 28 29 3b 0a 09 likes.size();..
0900: 09 66 6f 72 28 69 6e 74 20 4d 3d 31 3b 20 4d 3c .for(int M=1; M<
0910: 3d 6e 3b 20 2b 2b 4d 29 20 7b 0a 09 09 09 74 79 =n; ++M) {....ty
0920: 70 65 64 65 66 20 70 61 69 72 3c 69 6e 74 2c 69 pedef pair<int,i
0930: 6e 74 3e 20 56 65 72 74 3b 0a 09 09 09 61 75 74 nt> Vert;....aut
0940: 6f 5f 70 74 72 3c 20 4d 61 78 46 6c 6f 77 3c 56 o_ptr< MaxFlow<V
0950: 65 72 74 2c 20 69 6e 74 3e 20 3e 20 6d 66 28 6e ert, int> > mf(n
0960: 65 77 20 4d 61 78 46 6c 6f 77 3c 56 65 72 74 2c ew MaxFlow<Vert,
0970: 20 69 6e 74 3e 29 3b 0a 0a 09 09 09 66 6f 72 28 int>);.....for(
0980: 69 6e 74 20 69 3d 30 3b 20 69 3c 6e 3b 20 2b 2b int i=0; i<n; ++
0990: 69 29 20 7b 0a 09 09 09 09 6d 66 2d 3e 61 64 64 i) {.....mf->add
09a0: 45 64 67 65 28 56 65 72 74 28 30 2c 30 29 2c 20 Edge(Vert(0,0),
09b0: 56 65 72 74 28 31 2c 69 29 2c 20 4d 29 3b 20 2f Vert(1,i), M); /
09c0: 2f 20 73 72 63 20 4c 0a 09 09 09 09 6d 66 2d 3e / src L.....mf->
09d0: 61 64 64 45 64 67 65 28 56 65 72 74 28 31 2c 69 addEdge(Vert(1,i
09e0: 29 2c 20 56 65 72 74 28 32 2c 69 29 2c 20 4d 29 ), Vert(2,i), M)
09f0: 3b 20 2f 2f 20 4c 20 59 4c 0a 09 09 09 09 6d 66 ; // L YL.....mf
0a00: 2d 3e 61 64 64 45 64 67 65 28 56 65 72 74 28 31 ->addEdge(Vert(1
0a10: 2c 69 29 2c 20 56 65 72 74 28 33 2c 69 29 2c 20 ,i), Vert(3,i),
0a20: 6b 29 3b 20 2f 2f 20 4c 20 4e 4c 0a 09 09 09 09 k); // L NL.....
0a30: 6d 66 2d 3e 61 64 64 45 64 67 65 28 56 65 72 74 mf->addEdge(Vert
0a40: 28 34 2c 69 29 2c 20 56 65 72 74 28 36 2c 69 29 (4,i), Vert(6,i)
0a50: 2c 20 4d 29 3b 20 2f 2f 20 59 52 20 52 0a 09 09 , M); // YR R...
0a60: 09 09 6d 66 2d 3e 61 64 64 45 64 67 65 28 56 65 ..mf->addEdge(Ve
0a70: 72 74 28 35 2c 69 29 2c 20 56 65 72 74 28 36 2c rt(5,i), Vert(6,
0a80: 69 29 2c 20 6b 29 3b 20 2f 2f 20 4e 52 20 52 0a i), k); // NR R.
0a90: 09 09 09 09 6d 66 2d 3e 61 64 64 45 64 67 65 28 ....mf->addEdge(
0aa0: 56 65 72 74 28 36 2c 69 29 2c 20 56 65 72 74 28 Vert(6,i), Vert(
0ab0: 37 2c 30 29 2c 20 4d 29 3b 20 2f 2f 20 52 20 64 7,0), M); // R d
0ac0: 73 74 0a 0a 09 09 09 09 66 6f 72 28 69 6e 74 20 st......for(int
0ad0: 6a 3d 30 3b 20 6a 3c 6e 3b 20 2b 2b 6a 29 0a 09 j=0; j<n; ++j)..
0ae0: 09 09 09 09 69 66 28 20 6c 69 6b 65 73 5b 69 5d ....if( likes[i]
0af0: 5b 6a 5d 20 3d 3d 20 27 59 27 20 29 0a 09 09 09 [j] == 'Y' )....
0b00: 09 09 09 6d 66 2d 3e 61 64 64 45 64 67 65 28 56 ...mf->addEdge(V
0b10: 65 72 74 28 32 2c 69 29 2c 20 56 65 72 74 28 34 ert(2,i), Vert(4
0b20: 2c 6a 29 2c 20 31 29 3b 0a 09 09 09 09 09 65 6c ,j), 1);......el
0b30: 73 65 0a 09 09 09 09 09 09 6d 66 2d 3e 61 64 64 se.......mf->add
0b40: 45 64 67 65 28 56 65 72 74 28 33 2c 69 29 2c 20 Edge(Vert(3,i),
0b50: 56 65 72 74 28 35 2c 6a 29 2c 20 31 29 3b 0a 09 Vert(5,j), 1);..
0b60: 09 09 7d 0a 0a 09 09 09 69 66 28 20 6d 66 2d 3e ..}.....if( mf->
0b70: 63 61 6c 63 28 56 65 72 74 28 30 2c 30 29 2c 20 calc(Vert(0,0),
0b80: 56 65 72 74 28 37 2c 30 29 29 20 3c 20 4d 2a 6e Vert(7,0)) < M*n
0b90: 20 29 0a 09 09 09 09 72 65 74 75 72 6e 20 4d 2d ).....return M-
0ba0: 31 3b 0a 09 09 7d 0a 09 09 72 65 74 75 72 6e 20 1;...}...return
0bb0: 6e 3b 0a 09 7d 0a 7d 3b 0a 0a 2f 2f 20 42 45 47 n;..}.};..// BEG
0bc0: 49 4e 20 43 55 54 20 48 45 52 45 0a 23 69 6e 63 IN CUT HERE.#inc
0bd0: 6c 75 64 65 20 3c 63 74 69 6d 65 3e 0a 64 6f 75 lude <ctime>.dou
0be0: 62 6c 65 20 73 74 61 72 74 5f 74 69 6d 65 3b 20 ble start_time;
0bf0: 73 74 72 69 6e 67 20 74 69 6d 65 72 28 29 0a 20 string timer().
0c00: 7b 20 6f 73 74 72 69 6e 67 73 74 72 65 61 6d 20 { ostringstream
0c10: 6f 73 3b 20 6f 73 20 3c 3c 20 22 20 28 22 20 3c os; os << " (" <
0c20: 3c 20 69 6e 74 28 28 63 6c 6f 63 6b 28 29 2d 73 < int((clock()-s
0c30: 74 61 72 74 5f 74 69 6d 65 29 2f 43 4c 4f 43 4b tart_time)/CLOCK
0c40: 53 5f 50 45 52 5f 53 45 43 2a 31 30 30 30 29 20 S_PER_SEC*1000)
0c50: 3c 3c 20 22 20 6d 73 65 63 29 22 3b 20 72 65 74 << " msec)"; ret
0c60: 75 72 6e 20 6f 73 2e 73 74 72 28 29 3b 20 7d 0a urn os.str(); }.
0c70: 74 65 6d 70 6c 61 74 65 3c 74 79 70 65 6e 61 6d template<typenam
0c80: 65 20 54 3e 20 6f 73 74 72 65 61 6d 26 20 6f 70 e T> ostream& op
0c90: 65 72 61 74 6f 72 3c 3c 28 6f 73 74 72 65 61 6d erator<<(ostream
0ca0: 26 20 6f 73 2c 20 63 6f 6e 73 74 20 76 65 63 74 & os, const vect
0cb0: 6f 72 3c 54 3e 26 20 76 29 0a 20 7b 20 6f 73 20 or<T>& v). { os
0cc0: 3c 3c 20 22 7b 20 22 3b 0a 20 20 20 66 6f 72 28 << "{ ";. for(
0cd0: 74 79 70 65 6e 61 6d 65 20 76 65 63 74 6f 72 3c typename vector<
0ce0: 54 3e 3a 3a 63 6f 6e 73 74 5f 69 74 65 72 61 74 T>::const_iterat
0cf0: 6f 72 20 69 74 3d 76 2e 62 65 67 69 6e 28 29 3b or it=v.begin();
0d00: 20 69 74 21 3d 76 2e 65 6e 64 28 29 3b 20 2b 2b it!=v.end(); ++
0d10: 69 74 29 0a 20 20 20 6f 73 20 3c 3c 20 27 5c 22 it). os << '\"
0d20: 27 20 3c 3c 20 2a 69 74 20 3c 3c 20 27 5c 22 27 ' << *it << '\"'
0d30: 20 3c 3c 20 28 69 74 2b 31 3d 3d 76 2e 65 6e 64 << (it+1==v.end
0d40: 28 29 20 3f 20 22 22 20 3a 20 22 2c 20 22 29 3b () ? "" : ", ");
0d50: 20 6f 73 20 3c 3c 20 22 20 7d 22 3b 20 72 65 74 os << " }"; ret
0d60: 75 72 6e 20 6f 73 3b 20 7d 0a 76 6f 69 64 20 76 urn os; }.void v
0d70: 65 72 69 66 79 5f 63 61 73 65 28 63 6f 6e 73 74 erify_case(const
0d80: 20 69 6e 74 26 20 45 78 70 65 63 74 65 64 2c 20 int& Expected,
0d90: 63 6f 6e 73 74 20 69 6e 74 26 20 52 65 63 65 69 const int& Recei
0da0: 76 65 64 29 20 7b 0a 20 62 6f 6f 6c 20 6f 6b 20 ved) {. bool ok
0db0: 3d 20 28 45 78 70 65 63 74 65 64 20 3d 3d 20 52 = (Expected == R
0dc0: 65 63 65 69 76 65 64 29 3b 0a 20 69 66 28 6f 6b eceived);. if(ok
0dd0: 29 20 63 65 72 72 20 3c 3c 20 22 50 41 53 53 45 ) cerr << "PASSE
0de0: 44 22 20 3c 3c 20 74 69 6d 65 72 28 29 20 3c 3c D" << timer() <<
0df0: 20 65 6e 64 6c 3b 20 20 65 6c 73 65 20 7b 20 63 endl; else { c
0e00: 65 72 72 20 3c 3c 20 22 46 41 49 4c 45 44 22 20 err << "FAILED"
0e10: 3c 3c 20 74 69 6d 65 72 28 29 20 3c 3c 20 65 6e << timer() << en
0e20: 64 6c 3b 0a 20 63 65 72 72 20 3c 3c 20 22 5c 74 dl;. cerr << "\t
0e30: 6f 3a 20 5c 22 22 20 3c 3c 20 45 78 70 65 63 74 o: \"" << Expect
0e40: 65 64 20 3c 3c 20 27 5c 22 27 20 3c 3c 20 65 6e ed << '\"' << en
0e50: 64 6c 20 3c 3c 20 22 5c 74 78 3a 20 5c 22 22 20 dl << "\tx: \""
0e60: 3c 3c 20 52 65 63 65 69 76 65 64 20 3c 3c 20 27 << Received << '
0e70: 5c 22 27 20 3c 3c 20 65 6e 64 6c 3b 20 7d 20 7d \"' << endl; } }
0e80: 0a 23 64 65 66 69 6e 65 20 43 41 53 45 28 4e 29 .#define CASE(N)
0e90: 20 7b 63 65 72 72 20 3c 3c 20 22 54 65 73 74 20 {cerr << "Test
0ea0: 43 61 73 65 20 23 22 20 3c 3c 20 4e 20 3c 3c 20 Case #" << N <<
0eb0: 22 2e 2e 2e 22 20 3c 3c 20 66 6c 75 73 68 3b 20 "..." << flush;
0ec0: 73 74 61 72 74 5f 74 69 6d 65 3d 63 6c 6f 63 6b start_time=clock
0ed0: 28 29 3b 0a 23 64 65 66 69 6e 65 20 45 4e 44 09 ();.#define END.
0ee0: 20 76 65 72 69 66 79 5f 63 61 73 65 28 5f 2c 20 verify_case(_,
0ef0: 44 61 6e 63 69 6e 67 50 61 72 74 79 28 29 2e 6d DancingParty().m
0f00: 61 78 44 61 6e 63 65 73 28 6c 69 6b 65 73 2c 20 axDances(likes,
0f10: 6b 29 29 3b 7d 0a 69 6e 74 20 6d 61 69 6e 28 29 k));}.int main()
0f20: 7b 0a 0a 43 41 53 45 28 30 29 0a 09 73 74 72 69 {..CASE(0)..stri
0f30: 6e 67 20 6c 69 6b 65 73 5f 5b 5d 20 3d 20 7b 22 ng likes_[] = {"
0f40: 59 59 59 22 2c 20 22 59 59 59 22 2c 20 22 59 59 YYY", "YYY", "YY
0f50: 59 22 7d 3b 0a 09 20 20 76 65 63 74 6f 72 20 3c Y"};.. vector <
0f60: 73 74 72 69 6e 67 3e 20 6c 69 6b 65 73 28 6c 69 string> likes(li
0f70: 6b 65 73 5f 2c 20 6c 69 6b 65 73 5f 2b 73 69 7a kes_, likes_+siz
0f80: 65 6f 66 28 6c 69 6b 65 73 5f 29 2f 73 69 7a 65 eof(likes_)/size
0f90: 6f 66 28 2a 6c 69 6b 65 73 5f 29 29 3b 20 0a 09 of(*likes_)); ..
0fa0: 69 6e 74 20 6b 20 3d 20 30 3b 20 0a 09 69 6e 74 int k = 0; ..int
0fb0: 20 5f 20 3d 20 33 3b 20 0a 45 4e 44 0a 43 41 53 _ = 3; .END.CAS
0fc0: 45 28 31 29 0a 09 73 74 72 69 6e 67 20 6c 69 6b E(1)..string lik
0fd0: 65 73 5f 5b 5d 20 3d 20 7b 22 59 59 59 22 2c 20 es_[] = {"YYY",
0fe0: 22 59 59 4e 22 2c 20 22 59 4e 59 22 7d 3b 0a 09 "YYN", "YNY"};..
0ff0: 20 20 76 65 63 74 6f 72 20 3c 73 74 72 69 6e 67 vector <string
1000: 3e 20 6c 69 6b 65 73 28 6c 69 6b 65 73 5f 2c 20 > likes(likes_,
1010: 6c 69 6b 65 73 5f 2b 73 69 7a 65 6f 66 28 6c 69 likes_+sizeof(li
1020: 6b 65 73 5f 29 2f 73 69 7a 65 6f 66 28 2a 6c 69 kes_)/sizeof(*li
1030: 6b 65 73 5f 29 29 3b 20 0a 09 69 6e 74 20 6b 20 kes_)); ..int k
1040: 3d 20 30 3b 20 0a 09 69 6e 74 20 5f 20 3d 20 32 = 0; ..int _ = 2
1050: 3b 20 0a 45 4e 44 0a 43 41 53 45 28 32 29 0a 09 ; .END.CASE(2)..
1060: 73 74 72 69 6e 67 20 6c 69 6b 65 73 5f 5b 5d 20 string likes_[]
1070: 3d 20 7b 22 59 4e 22 2c 20 22 59 4e 22 7d 3b 0a = {"YN", "YN"};.
1080: 09 20 20 76 65 63 74 6f 72 20 3c 73 74 72 69 6e . vector <strin
1090: 67 3e 20 6c 69 6b 65 73 28 6c 69 6b 65 73 5f 2c g> likes(likes_,
10a0: 20 6c 69 6b 65 73 5f 2b 73 69 7a 65 6f 66 28 6c likes_+sizeof(l
10b0: 69 6b 65 73 5f 29 2f 73 69 7a 65 6f 66 28 2a 6c ikes_)/sizeof(*l
10c0: 69 6b 65 73 5f 29 29 3b 20 0a 09 69 6e 74 20 6b ikes_)); ..int k
10d0: 20 3d 20 30 3b 20 0a 09 69 6e 74 20 5f 20 3d 20 = 0; ..int _ =
10e0: 30 3b 20 0a 45 4e 44 0a 43 41 53 45 28 33 29 0a 0; .END.CASE(3).
10f0: 09 73 74 72 69 6e 67 20 6c 69 6b 65 73 5f 5b 5d .string likes_[]
1100: 20 3d 20 7b 22 59 4e 22 2c 20 22 59 4e 22 7d 3b = {"YN", "YN"};
1110: 0a 09 20 20 76 65 63 74 6f 72 20 3c 73 74 72 69 .. vector <stri
1120: 6e 67 3e 20 6c 69 6b 65 73 28 6c 69 6b 65 73 5f ng> likes(likes_
1130: 2c 20 6c 69 6b 65 73 5f 2b 73 69 7a 65 6f 66 28 , likes_+sizeof(
1140: 6c 69 6b 65 73 5f 29 2f 73 69 7a 65 6f 66 28 2a likes_)/sizeof(*
1150: 6c 69 6b 65 73 5f 29 29 3b 20 0a 09 69 6e 74 20 likes_)); ..int
1160: 6b 20 3d 20 31 3b 20 0a 09 69 6e 74 20 5f 20 3d k = 1; ..int _ =
1170: 20 31 3b 20 0a 45 4e 44 0a 2f 2a 0a 43 41 53 45 1; .END./*.CASE
1180: 28 34 29 0a 09 73 74 72 69 6e 67 20 6c 69 6b 65 (4)..string like
1190: 73 5f 5b 5d 20 3d 20 3b 0a 09 20 20 76 65 63 74 s_[] = ;.. vect
11a0: 6f 72 20 3c 73 74 72 69 6e 67 3e 20 6c 69 6b 65 or <string> like
11b0: 73 28 6c 69 6b 65 73 5f 2c 20 6c 69 6b 65 73 5f s(likes_, likes_
11c0: 2b 73 69 7a 65 6f 66 28 6c 69 6b 65 73 5f 29 2f +sizeof(likes_)/
11d0: 73 69 7a 65 6f 66 28 2a 6c 69 6b 65 73 5f 29 29 sizeof(*likes_))
11e0: 3b 20 0a 09 69 6e 74 20 6b 20 3d 20 3b 20 0a 09 ; ..int k = ; ..
11f0: 69 6e 74 20 5f 20 3d 20 3b 20 0a 45 4e 44 0a 43 int _ = ; .END.C
1200: 41 53 45 28 35 29 0a 09 73 74 72 69 6e 67 20 6c ASE(5)..string l
1210: 69 6b 65 73 5f 5b 5d 20 3d 20 3b 0a 09 20 20 76 ikes_[] = ;.. v
1220: 65 63 74 6f 72 20 3c 73 74 72 69 6e 67 3e 20 6c ector <string> l
1230: 69 6b 65 73 28 6c 69 6b 65 73 5f 2c 20 6c 69 6b ikes(likes_, lik
1240: 65 73 5f 2b 73 69 7a 65 6f 66 28 6c 69 6b 65 73 es_+sizeof(likes
1250: 5f 29 2f 73 69 7a 65 6f 66 28 2a 6c 69 6b 65 73 _)/sizeof(*likes
1260: 5f 29 29 3b 20 0a 09 69 6e 74 20 6b 20 3d 20 3b _)); ..int k = ;
1270: 20 0a 09 69 6e 74 20 5f 20 3d 20 3b 20 0a 45 4e ..int _ = ; .EN
1280: 44 0a 2a 2f 0a 7d 0a 2f 2f 20 45 4e 44 20 43 55 D.*/.}.// END CU
1290: 54 20 48 45 52 45 0a T HERE.