Artifact 296b9275312ebbf1e64a9e94d6e99fdf47d1d6c2:
0000: 0a 2f 2f 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d .//-------------
0010: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
0020: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
0030: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
0040: 0a 2f 2f 20 44 69 6e 69 63 27 73 20 41 6c 67 6f .// Dinic's Algo
0050: 72 69 74 68 6d 0a 2f 2f 20 20 20 4f 28 56 20 45 rithm.// O(V E
0060: 29 0a 2f 2f 0a 2f 2f 20 47 20 3a 20 62 69 64 69 ).//.// G : bidi
0070: 72 65 63 74 69 6f 6e 61 6c 20 28 47 5b 69 5d 2e rectional (G[i].
0080: 68 61 73 28 6a 29 20 3c 3d 3d 3e 20 47 5b 6a 5d has(j) <==> G[j]
0090: 2e 68 61 73 28 69 29 29 0a 2f 2f 20 46 20 3a 20 .has(i)).// F :
00a0: 66 6c 6f 77 2d 63 61 70 61 63 69 74 79 20 46 5b flow-capacity F[
00b0: 69 5d 5b 6a 5d 20 3d 20 43 61 70 61 63 69 74 79 i][j] = Capacity
00c0: 2c 20 46 5b 6a 5d 5b 69 5d 20 3d 20 30 0a 2f 2f , F[j][i] = 0.//
00d0: 0a 2f 2f 20 56 65 72 69 66 69 65 64 20 62 79 0a .// Verified by.
00e0: 2f 2f 20 20 20 2d 20 53 52 4d 20 33 39 39 20 44 // - SRM 399 D
00f0: 69 76 31 20 4c 56 33 0a 2f 2f 20 20 20 2d 20 50 iv1 LV3.// - P
0100: 4b 55 20 31 34 35 39 0a 2f 2f 20 20 20 2d 20 43 KU 1459.// - C
0110: 6f 64 65 43 72 61 66 74 20 30 39 20 43 55 54 53 odeCraft 09 CUTS
0120: 0a 2f 2f 20 20 20 2d 20 53 52 4d 20 34 36 35 20 .// - SRM 465
0130: 44 69 76 31 20 4c 56 32 0a 2f 2f 2d 2d 2d 2d 2d Div1 LV2.//-----
0140: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
0150: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
0160: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
0170: 2d 2d 2d 2d 2d 2d 2d 2d 0a 0a 74 65 6d 70 6c 61 --------..templa
0180: 74 65 3c 74 79 70 65 6e 61 6d 65 20 54 3e 0a 63 te<typename T>.c
0190: 6c 61 73 73 20 49 64 47 65 6e 0a 7b 0a 09 6d 61 lass IdGen.{..ma
01a0: 70 3c 54 2c 20 69 6e 74 3e 20 76 32 69 64 5f 3b p<T, int> v2id_;
01b0: 0a 09 76 65 63 74 6f 72 3c 54 3e 20 20 20 69 64 ..vector<T> id
01c0: 32 76 5f 3b 0a 70 75 62 6c 69 63 3a 0a 09 69 6e 2v_;.public:..in
01d0: 74 20 76 32 69 64 28 63 6f 6e 73 74 20 54 26 20 t v2id(const T&
01e0: 76 29 20 7b 0a 09 09 69 66 28 20 21 76 32 69 64 v) {...if( !v2id
01f0: 5f 2e 63 6f 75 6e 74 28 76 29 20 29 20 7b 20 76 _.count(v) ) { v
0200: 32 69 64 5f 5b 76 5d 20 3d 20 73 69 7a 65 28 29 2id_[v] = size()
0210: 3b 20 69 64 32 76 5f 2e 70 75 73 68 5f 62 61 63 ; id2v_.push_bac
0220: 6b 28 76 29 3b 20 7d 0a 09 09 72 65 74 75 72 6e k(v); }...return
0230: 20 76 32 69 64 5f 5b 76 5d 3b 0a 09 7d 0a 09 63 v2id_[v];..}..c
0240: 6f 6e 73 74 20 54 26 20 69 64 32 76 28 69 6e 74 onst T& id2v(int
0250: 20 69 29 20 63 6f 6e 73 74 20 7b 20 72 65 74 75 i) const { retu
0260: 72 6e 20 69 64 32 76 5f 5b 69 5d 3b 20 7d 0a 09 rn id2v_[i]; }..
0270: 69 6e 74 20 73 69 7a 65 28 29 20 63 6f 6e 73 74 int size() const
0280: 20 7b 20 72 65 74 75 72 6e 20 69 64 32 76 5f 2e { return id2v_.
0290: 73 69 7a 65 28 29 3b 20 7d 0a 7d 3b 0a 0a 74 65 size(); }.};..te
02a0: 6d 70 6c 61 74 65 3c 74 79 70 65 6e 61 6d 65 20 mplate<typename
02b0: 56 65 72 74 2c 20 74 79 70 65 6e 61 6d 65 20 46 Vert, typename F
02c0: 6c 6f 77 2c 20 69 6e 74 20 4e 56 3d 35 31 32 3e low, int NV=512>
02d0: 0a 63 6c 61 73 73 20 4d 61 78 46 6c 6f 77 0a 7b .class MaxFlow.{
02e0: 0a 09 49 64 47 65 6e 3c 56 65 72 74 3e 20 69 64 ..IdGen<Vert> id
02f0: 67 65 6e 3b 0a 09 76 65 63 74 6f 72 3c 69 6e 74 gen;..vector<int
0300: 3e 20 47 5b 4e 56 5d 3b 0a 09 46 6c 6f 77 20 46 > G[NV];..Flow F
0310: 5b 4e 56 5d 5b 4e 56 5d 3b 0a 0a 70 75 62 6c 69 [NV][NV];..publi
0320: 63 3a 0a 09 76 6f 69 64 20 61 64 64 45 64 67 65 c:..void addEdge
0330: 28 20 56 65 72 74 20 73 5f 2c 20 56 65 72 74 20 ( Vert s_, Vert
0340: 74 5f 2c 20 46 6c 6f 77 20 66 20 29 0a 09 7b 0a t_, Flow f )..{.
0350: 09 09 63 6f 6e 73 74 20 69 6e 74 20 73 20 3d 20 ..const int s =
0360: 69 64 67 65 6e 2e 76 32 69 64 28 73 5f 29 2c 20 idgen.v2id(s_),
0370: 74 20 3d 20 69 64 67 65 6e 2e 76 32 69 64 28 74 t = idgen.v2id(t
0380: 5f 29 3b 0a 09 09 47 5b 73 5d 2e 70 75 73 68 5f _);...G[s].push_
0390: 62 61 63 6b 28 74 29 3b 0a 09 09 47 5b 74 5d 2e back(t);...G[t].
03a0: 70 75 73 68 5f 62 61 63 6b 28 73 29 3b 0a 09 09 push_back(s);...
03b0: 46 5b 73 5d 5b 74 5d 20 3d 20 66 3b 0a 09 09 46 F[s][t] = f;...F
03c0: 5b 74 5d 5b 73 5d 20 3d 20 30 3b 0a 09 7d 0a 0a [t][s] = 0;..}..
03d0: 09 46 6c 6f 77 20 63 61 6c 63 28 20 56 65 72 74 .Flow calc( Vert
03e0: 20 73 5f 2c 20 56 65 72 74 20 74 5f 20 29 0a 09 s_, Vert t_ )..
03f0: 7b 0a 09 09 63 6f 6e 73 74 20 69 6e 74 20 53 20 {...const int S
0400: 3d 20 69 64 67 65 6e 2e 76 32 69 64 28 73 5f 29 = idgen.v2id(s_)
0410: 2c 20 44 20 3d 20 69 64 67 65 6e 2e 76 32 69 64 , D = idgen.v2id
0420: 28 74 5f 29 3b 0a 09 09 66 6f 72 28 20 46 6c 6f (t_);...for( Flo
0430: 77 20 74 6f 74 61 6c 3d 30 20 3b 3b 20 29 20 7b w total=0 ;; ) {
0440: 0a 09 09 09 2f 2f 20 44 6f 20 42 46 53 20 61 6e ....// Do BFS an
0450: 64 20 63 6f 6d 70 75 74 65 20 74 68 65 20 6c 65 d compute the le
0460: 76 65 6c 20 66 6f 72 20 65 61 63 68 20 6e 6f 64 vel for each nod
0470: 65 2e 0a 09 09 09 69 6e 74 20 4c 56 5b 4e 56 5d e.....int LV[NV]
0480: 20 3d 20 7b 30 7d 3b 0a 09 09 09 76 65 63 74 6f = {0};....vecto
0490: 72 3c 69 6e 74 3e 20 51 28 31 2c 20 53 29 3b 0a r<int> Q(1, S);.
04a0: 09 09 09 66 6f 72 28 69 6e 74 20 6c 76 3d 31 3b ...for(int lv=1;
04b0: 20 21 51 2e 65 6d 70 74 79 28 29 3b 20 2b 2b 6c !Q.empty(); ++l
04c0: 76 29 20 7b 0a 09 09 09 09 76 65 63 74 6f 72 3c v) {.....vector<
04d0: 69 6e 74 3e 20 51 32 3b 0a 09 09 09 09 66 6f 72 int> Q2;.....for
04e0: 28 73 69 7a 65 5f 74 20 69 3d 30 3b 20 69 21 3d (size_t i=0; i!=
04f0: 51 2e 73 69 7a 65 28 29 3b 20 2b 2b 69 29 20 7b Q.size(); ++i) {
0500: 0a 09 09 09 09 09 63 6f 6e 73 74 20 76 65 63 74 ......const vect
0510: 6f 72 3c 69 6e 74 3e 26 20 6e 65 20 3d 20 47 5b or<int>& ne = G[
0520: 51 5b 69 5d 5d 3b 0a 09 09 09 09 09 66 6f 72 28 Q[i]];......for(
0530: 73 69 7a 65 5f 74 20 6a 3d 30 3b 20 6a 21 3d 6e size_t j=0; j!=n
0540: 65 2e 73 69 7a 65 28 29 3b 20 2b 2b 6a 29 0a 09 e.size(); ++j)..
0550: 09 09 09 09 09 69 66 28 20 46 5b 51 5b 69 5d 5d .....if( F[Q[i]]
0560: 5b 6e 65 5b 6a 5d 5d 20 26 26 20 21 4c 56 5b 6e [ne[j]] && !LV[n
0570: 65 5b 6a 5d 5d 20 26 26 20 6e 65 5b 6a 5d 21 3d e[j]] && ne[j]!=
0580: 53 20 29 0a 09 09 09 09 09 09 09 4c 56 5b 6e 65 S )........LV[ne
0590: 5b 6a 5d 5d 3d 6c 76 2c 20 51 32 2e 70 75 73 68 [j]]=lv, Q2.push
05a0: 5f 62 61 63 6b 28 6e 65 5b 6a 5d 29 3b 0a 09 09 _back(ne[j]);...
05b0: 09 09 7d 0a 09 09 09 09 51 2e 73 77 61 70 28 51 ..}.....Q.swap(Q
05c0: 32 29 3b 0a 09 09 09 7d 0a 0a 09 09 09 2f 2f 20 2);....}.....//
05d0: 44 65 73 74 69 6e 61 74 69 6f 6e 20 69 73 20 6e Destination is n
05e0: 6f 77 20 75 6e 72 65 61 63 68 61 62 6c 65 2e 20 ow unreachable.
05f0: 44 6f 6e 65 2e 0a 09 09 09 69 66 28 20 21 4c 56 Done.....if( !LV
0600: 5b 44 5d 20 29 0a 09 09 09 09 72 65 74 75 72 6e [D] ).....return
0610: 20 74 6f 74 61 6c 3b 0a 0a 09 09 09 2f 2f 20 49 total;.....// I
0620: 74 65 72 61 74 69 6e 67 20 44 46 53 2e 0a 09 09 terating DFS....
0630: 09 62 6f 6f 6c 20 62 6c 6f 63 6b 65 64 5b 4e 56 .bool blocked[NV
0640: 5d 20 3d 20 7b 7d 3b 0a 09 09 09 74 6f 74 61 6c ] = {};....total
0650: 20 2b 3d 20 64 69 6e 69 63 5f 64 66 73 28 20 53 += dinic_dfs( S
0660: 2c 20 44 2c 20 4c 56 2c 20 30 78 37 66 66 66 66 , D, LV, 0x7ffff
0670: 66 66 66 2c 20 62 6c 6f 63 6b 65 64 20 29 3b 0a fff, blocked );.
0680: 09 09 7d 0a 09 7d 0a 0a 70 72 69 76 61 74 65 3a ..}..}..private:
0690: 0a 09 46 6c 6f 77 20 64 69 6e 69 63 5f 64 66 73 ..Flow dinic_dfs
06a0: 28 20 69 6e 74 20 76 2c 20 69 6e 74 20 44 2c 20 ( int v, int D,
06b0: 69 6e 74 20 4c 56 5b 5d 2c 20 46 6c 6f 77 20 66 int LV[], Flow f
06c0: 6c 6f 77 5f 69 6e 2c 20 62 6f 6f 6c 20 62 6c 6f low_in, bool blo
06d0: 63 6b 65 64 5b 5d 20 29 0a 09 7b 0a 09 09 46 6c cked[] )..{...Fl
06e0: 6f 77 20 66 6c 6f 77 5f 6f 75 74 20 3d 20 30 3b ow flow_out = 0;
06f0: 0a 09 09 66 6f 72 28 73 69 7a 65 5f 74 20 69 3d ...for(size_t i=
0700: 30 3b 20 69 21 3d 47 5b 76 5d 2e 73 69 7a 65 28 0; i!=G[v].size(
0710: 29 3b 20 2b 2b 69 29 20 7b 0a 09 09 09 69 6e 74 ); ++i) {....int
0720: 20 75 20 3d 20 47 5b 76 5d 5b 69 5d 3b 0a 09 09 u = G[v][i];...
0730: 09 69 66 28 20 4c 56 5b 76 5d 2b 31 3d 3d 4c 56 .if( LV[v]+1==LV
0740: 5b 75 5d 20 26 26 20 46 5b 76 5d 5b 75 5d 20 29 [u] && F[v][u] )
0750: 20 7b 0a 09 09 09 09 46 6c 6f 77 20 66 20 3d 20 {.....Flow f =
0760: 6d 69 6e 28 66 6c 6f 77 5f 69 6e 2d 66 6c 6f 77 min(flow_in-flow
0770: 5f 6f 75 74 2c 20 46 5b 76 5d 5b 75 5d 29 3b 0a _out, F[v][u]);.
0780: 09 09 09 09 69 66 28 20 75 3d 3d 44 20 7c 7c 20 ....if( u==D ||
0790: 21 62 6c 6f 63 6b 65 64 5b 75 5d 20 26 26 20 28 !blocked[u] && (
07a0: 66 3d 64 69 6e 69 63 5f 64 66 73 28 75 2c 44 2c f=dinic_dfs(u,D,
07b0: 4c 56 2c 66 2c 62 6c 6f 63 6b 65 64 29 29 3e 30 LV,f,blocked))>0
07c0: 20 29 20 7b 0a 09 09 09 09 09 46 5b 76 5d 5b 75 ) {......F[v][u
07d0: 5d 20 20 2d 3d 20 66 3b 0a 09 09 09 09 09 46 5b ] -= f;......F[
07e0: 75 5d 5b 76 5d 20 20 2b 3d 20 66 3b 0a 09 09 09 u][v] += f;....
07f0: 09 09 66 6c 6f 77 5f 6f 75 74 20 2b 3d 20 66 3b ..flow_out += f;
0800: 0a 09 09 09 09 09 69 66 28 20 66 6c 6f 77 5f 69 ......if( flow_i
0810: 6e 20 3d 3d 20 66 6c 6f 77 5f 6f 75 74 20 29 20 n == flow_out )
0820: 72 65 74 75 72 6e 20 66 6c 6f 77 5f 6f 75 74 3b return flow_out;
0830: 0a 09 09 09 09 7d 0a 09 09 09 7d 0a 09 09 7d 0a .....}....}...}.
0840: 09 09 62 6c 6f 63 6b 65 64 5b 76 5d 20 3d 20 28 ..blocked[v] = (
0850: 66 6c 6f 77 5f 6f 75 74 3d 3d 30 29 3b 0a 09 09 flow_out==0);...
0860: 72 65 74 75 72 6e 20 66 6c 6f 77 5f 6f 75 74 3b return flow_out;
0870: 0a 09 7d 0a 7d 3b 0a ..}.};.