Artifact d90f43e790c04822abf6f6e89ecea9c73383b1f7:
0000: 2f 2f 2d 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 0a ---------------.
0040: 2f 2f 20 53 65 67 6d 65 6e 74 20 74 72 65 65 0a // Segment tree.
0050: 2f 2f 20 20 20 69 6e 20 73 6f 6d 65 20 67 65 6e // in some gen
0060: 65 72 61 6c 20 66 6f 72 6d 0a 2f 2f 0a 2f 2f 20 eral form.//.//
0070: 56 65 72 69 66 69 65 64 20 62 79 0a 2f 2f 20 20 Verified by.//
0080: 20 2d 20 43 6f 64 65 66 6f 72 63 65 73 20 31 30 - Codeforces 10
0090: 34 20 45 0a 2f 2f 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 4 E.//----------
00a0: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
00b0: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
00c0: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
00d0: 2d 2d 2d 0a 0a 63 6c 61 73 73 20 53 65 67 6d 65 ---..class Segme
00e0: 6e 74 54 72 65 65 0a 7b 0a 09 2f 2f 20 2d 2d 2d ntTree.{..// ---
00f0: 20 44 41 54 41 20 46 4f 52 20 45 41 43 48 20 4e DATA FOR EACH N
0100: 4f 44 45 20 2d 2d 2d 0a 09 73 74 72 75 63 74 20 ODE ---..struct
0110: 4e 6f 64 65 0a 09 7b 0a 09 09 69 6e 74 20 73 75 Node..{...int su
0120: 6d 3b 0a 09 09 69 6e 74 20 6d 61 78 4c 65 66 74 m;...int maxLeft
0130: 3b 0a 09 09 69 6e 74 20 6d 69 6e 4c 65 66 74 3b ;...int minLeft;
0140: 0a 09 09 69 6e 74 20 6e 34 3b 0a 09 09 69 6e 74 ...int n4;...int
0150: 20 6e 37 3b 0a 09 09 62 6f 6f 6c 20 73 77 69 74 n7;...bool swit
0160: 63 68 65 64 3b 0a 09 7d 3b 0a 09 2f 2f 20 2d 2d ched;..};..// --
0170: 2d 20 44 41 54 41 20 46 4f 52 20 45 41 43 48 20 - DATA FOR EACH
0180: 4e 4f 44 45 20 2d 2d 2d 0a 0a 09 4e 6f 64 65 20 NODE ---...Node
0190: 43 61 6e 6f 6e 69 63 61 6c 28 69 6e 74 20 6c 76 Canonical(int lv
01a0: 2c 20 69 6e 74 20 69 64 78 29 0a 09 7b 0a 09 09 , int idx)..{...
01b0: 4e 6f 64 65 26 20 78 20 3d 20 74 72 65 65 5b 6c Node& x = tree[l
01c0: 76 5d 5b 69 64 78 5d 3b 0a 09 09 2f 2f 20 2d 2d v][idx];...// --
01d0: 2d 20 44 4f 20 53 4f 4d 45 54 48 49 4e 47 20 49 - DO SOMETHING I
01e0: 46 20 59 4f 55 20 4e 45 45 44 20 43 41 4e 4f 4e F YOU NEED CANON
01f0: 49 43 41 4c 49 5a 41 54 49 4f 4e 20 2d 2d 2d 0a ICALIZATION ---.
0200: 09 09 69 66 28 20 21 78 2e 73 77 69 74 63 68 65 ..if( !x.switche
0210: 64 20 29 0a 09 09 09 72 65 74 75 72 6e 20 78 3b d )....return x;
0220: 0a 09 09 4e 6f 64 65 20 6e 20 3d 20 7b 2d 78 2e ...Node n = {-x.
0230: 73 75 6d 2c 20 2d 78 2e 6d 69 6e 4c 65 66 74 2c sum, -x.minLeft,
0240: 20 2d 78 2e 6d 61 78 4c 65 66 74 2c 20 78 2e 6e -x.maxLeft, x.n
0250: 37 2c 20 78 2e 6e 34 2c 20 66 61 6c 73 65 7d 3b 7, x.n4, false};
0260: 0a 09 09 72 65 74 75 72 6e 20 6e 3b 0a 09 09 2f ...return n;.../
0270: 2f 20 2d 2d 2d 20 44 4f 20 53 4f 4d 45 54 48 49 / --- DO SOMETHI
0280: 4e 47 20 49 46 20 59 4f 55 20 4e 45 45 44 20 43 NG IF YOU NEED C
0290: 41 4e 4f 4e 49 43 41 4c 49 5a 41 54 49 4f 4e 20 ANONICALIZATION
02a0: 2d 2d 2d 0a 09 7d 0a 0a 09 76 6f 69 64 20 55 70 ---..}...void Up
02b0: 64 61 74 65 4d 69 64 4e 6f 64 65 28 69 6e 74 20 dateMidNode(int
02c0: 6c 76 2c 20 69 6e 74 20 69 64 78 29 0a 09 7b 0a lv, int idx)..{.
02d0: 09 09 4e 6f 64 65 20 6c 20 3d 20 43 61 6e 6f 6e ..Node l = Canon
02e0: 69 63 61 6c 28 6c 76 2d 31 2c 20 69 64 78 2a 32 ical(lv-1, idx*2
02f0: 29 3b 0a 09 09 4e 6f 64 65 20 72 20 3d 20 43 61 );...Node r = Ca
0300: 6e 6f 6e 69 63 61 6c 28 6c 76 2d 31 2c 20 69 64 nonical(lv-1, id
0310: 78 2a 32 2b 31 29 3b 0a 09 09 2f 2f 20 2d 2d 2d x*2+1);...// ---
0320: 20 42 4f 54 54 4f 4d 20 55 50 20 43 4f 4d 50 55 BOTTOM UP COMPU
0330: 54 41 54 49 4f 4e 20 2d 2d 2d 0a 09 09 4e 6f 64 TATION ---...Nod
0340: 65 20 6d 65 20 3d 20 7b 0a 09 09 09 6c 2e 73 75 e me = {....l.su
0350: 6d 20 2b 20 72 2e 73 75 6d 2c 0a 09 09 09 6d 61 m + r.sum,....ma
0360: 78 28 6c 2e 6d 61 78 4c 65 66 74 2c 20 6c 2e 73 x(l.maxLeft, l.s
0370: 75 6d 2b 72 2e 6d 61 78 4c 65 66 74 29 2c 0a 09 um+r.maxLeft),..
0380: 09 09 6d 69 6e 28 6c 2e 6d 69 6e 4c 65 66 74 2c ..min(l.minLeft,
0390: 20 6c 2e 73 75 6d 2b 72 2e 6d 69 6e 4c 65 66 74 l.sum+r.minLeft
03a0: 29 2c 0a 09 09 09 6c 2e 6e 34 20 2b 20 72 2e 6e ),....l.n4 + r.n
03b0: 34 2c 0a 09 09 09 6c 2e 6e 37 20 2b 20 72 2e 6e 4,....l.n7 + r.n
03c0: 37 2c 0a 09 09 09 66 61 6c 73 65 0a 09 09 7d 3b 7,....false...};
03d0: 0a 09 09 2f 2f 20 2d 2d 2d 20 42 4f 54 54 4f 4d ...// --- BOTTOM
03e0: 20 55 50 20 43 4f 4d 50 55 54 41 54 49 4f 4e 20 UP COMPUTATION
03f0: 2d 2d 2d 0a 09 09 74 72 65 65 5b 6c 76 5d 5b 69 ---...tree[lv][i
0400: 64 78 5d 20 3d 20 6d 65 3b 0a 09 7d 0a 0a 09 76 dx] = me;..}...v
0410: 65 63 74 6f 72 3c 20 76 65 63 74 6f 72 3c 4e 6f ector< vector<No
0420: 64 65 3e 20 3e 20 74 72 65 65 3b 0a 0a 70 75 62 de> > tree;..pub
0430: 6c 69 63 3a 0a 09 53 65 67 6d 65 6e 74 54 72 65 lic:..SegmentTre
0440: 65 28 63 6f 6e 73 74 20 73 74 72 69 6e 67 26 20 e(const string&
0450: 73 29 0a 09 7b 0a 09 09 69 6e 74 20 4e 20 3d 20 s)..{...int N =
0460: 31 3b 0a 09 09 77 68 69 6c 65 28 20 4e 20 3c 20 1;...while( N <
0470: 73 2e 73 69 7a 65 28 29 20 29 0a 09 09 09 4e 20 s.size() )....N
0480: 3c 3c 3d 20 31 3b 0a 0a 09 09 74 72 65 65 2e 72 <<= 1;....tree.r
0490: 65 73 69 7a 65 28 74 72 65 65 2e 73 69 7a 65 28 esize(tree.size(
04a0: 29 2b 31 29 3b 20 74 72 65 65 2e 62 61 63 6b 28 )+1); tree.back(
04b0: 29 2e 72 65 73 69 7a 65 28 4e 29 3b 0a 09 09 66 ).resize(N);...f
04c0: 6f 72 28 69 6e 74 20 69 3d 30 3b 20 69 3c 4e 3b or(int i=0; i<N;
04d0: 20 2b 2b 69 29 0a 09 09 7b 0a 09 09 09 2f 2f 20 ++i)...{....//
04e0: 2d 2d 2d 20 49 4e 49 54 49 41 4c 49 5a 45 20 42 --- INITIALIZE B
04f0: 41 53 45 20 31 2d 45 4c 45 4d 45 4e 54 20 4e 4f ASE 1-ELEMENT NO
0500: 44 45 53 20 2d 2d 2d 0a 09 09 09 69 6e 74 20 76 DES ---....int v
0510: 20 3d 20 69 3c 73 2e 73 69 7a 65 28 29 20 3f 20 = i<s.size() ?
0520: 28 73 5b 69 5d 3d 3d 27 34 27 20 3f 20 2b 31 20 (s[i]=='4' ? +1
0530: 3a 20 2d 31 29 20 3a 20 30 3b 0a 09 09 09 4e 6f : -1) : 0;....No
0540: 64 65 20 6d 65 20 3d 20 7b 76 2c 20 6d 61 78 28 de me = {v, max(
0550: 30 2c 76 29 2c 20 6d 69 6e 28 30 2c 76 29 2c 20 0,v), min(0,v),
0560: 76 3d 3d 2b 31 2c 20 76 3d 3d 2d 31 2c 20 66 61 v==+1, v==-1, fa
0570: 6c 73 65 7d 3b 0a 09 09 09 2f 2f 20 2d 2d 2d 20 lse};....// ---
0580: 49 4e 49 54 49 41 4c 49 5a 45 20 42 41 53 45 20 INITIALIZE BASE
0590: 31 2d 45 4c 45 4d 45 4e 54 20 4e 4f 44 45 53 20 1-ELEMENT NODES
05a0: 2d 2d 2d 0a 09 09 09 74 72 65 65 2e 62 61 63 6b ---....tree.back
05b0: 28 29 5b 69 5d 20 3d 20 6d 65 3b 0a 09 09 7d 0a ()[i] = me;...}.
05c0: 0a 09 09 77 68 69 6c 65 28 4e 3e 3e 3d 31 29 0a ...while(N>>=1).
05d0: 09 09 7b 0a 09 09 09 74 72 65 65 2e 72 65 73 69 ..{....tree.resi
05e0: 7a 65 28 74 72 65 65 2e 73 69 7a 65 28 29 2b 31 ze(tree.size()+1
05f0: 29 3b 20 74 72 65 65 2e 62 61 63 6b 28 29 2e 72 ); tree.back().r
0600: 65 73 69 7a 65 28 4e 29 3b 0a 09 09 09 66 6f 72 esize(N);....for
0610: 28 69 6e 74 20 69 3d 30 3b 20 69 3c 4e 3b 20 2b (int i=0; i<N; +
0620: 2b 69 29 0a 09 09 09 09 55 70 64 61 74 65 4d 69 +i).....UpdateMi
0630: 64 4e 6f 64 65 28 74 72 65 65 2e 73 69 7a 65 28 dNode(tree.size(
0640: 29 2d 31 2c 20 69 29 3b 0a 09 09 7d 0a 09 7d 0a )-1, i);...}..}.
0650: 0a 09 69 6e 74 20 43 6f 75 6e 74 28 29 0a 09 7b ..int Count()..{
0660: 0a 09 09 4e 6f 64 65 20 78 20 3d 20 43 61 6e 6f ...Node x = Cano
0670: 6e 69 63 61 6c 28 74 72 65 65 2e 73 69 7a 65 28 nical(tree.size(
0680: 29 2d 31 2c 20 30 29 3b 0a 09 09 72 65 74 75 72 )-1, 0);...retur
0690: 6e 20 78 2e 6d 61 78 4c 65 66 74 20 2b 20 78 2e n x.maxLeft + x.
06a0: 6e 37 3b 0a 09 7d 0a 0a 09 76 6f 69 64 20 53 77 n7;..}...void Sw
06b0: 69 74 63 68 28 69 6e 74 20 4c 2c 20 69 6e 74 20 itch(int L, int
06c0: 52 29 0a 09 7b 0a 09 09 53 77 69 74 63 68 52 65 R)..{...SwitchRe
06d0: 63 28 74 72 65 65 5b 30 5d 2e 73 69 7a 65 28 29 c(tree[0].size()
06e0: 2c 20 74 72 65 65 2e 73 69 7a 65 28 29 2d 31 2c , tree.size()-1,
06f0: 20 30 2c 20 4c 2c 20 52 29 3b 0a 09 7d 0a 0a 09 0, L, R);..}...
0700: 76 6f 69 64 20 53 77 69 74 63 68 52 65 63 28 69 void SwitchRec(i
0710: 6e 74 20 53 74 72 69 64 65 2c 20 69 6e 74 20 6c nt Stride, int l
0720: 76 2c 20 69 6e 74 20 69 64 78 2c 20 69 6e 74 20 v, int idx, int
0730: 52 61 77 5f 4c 2c 20 69 6e 74 20 52 61 77 5f 52 Raw_L, int Raw_R
0740: 29 0a 09 7b 0a 09 09 2f 2f 20 4f 75 74 73 69 64 )..{...// Outsid
0750: 65 0a 09 09 69 66 28 20 52 61 77 5f 52 20 3c 3d e...if( Raw_R <=
0760: 20 53 74 72 69 64 65 2a 69 64 78 20 7c 7c 20 53 Stride*idx || S
0770: 74 72 69 64 65 2a 28 69 64 78 2b 31 29 20 3c 3d tride*(idx+1) <=
0780: 20 52 61 77 5f 4c 20 29 0a 09 09 09 72 65 74 75 Raw_L )....retu
0790: 72 6e 3b 0a 0a 09 09 2f 2f 20 49 6e 73 69 64 65 rn;....// Inside
07a0: 0a 09 09 62 6f 6f 6c 26 20 6d 65 20 3d 20 74 72 ...bool& me = tr
07b0: 65 65 5b 6c 76 5d 5b 69 64 78 5d 2e 73 77 69 74 ee[lv][idx].swit
07c0: 63 68 65 64 3b 0a 09 09 69 66 28 20 52 61 77 5f ched;...if( Raw_
07d0: 4c 20 3c 3d 20 53 74 72 69 64 65 2a 69 64 78 20 L <= Stride*idx
07e0: 26 26 20 53 74 72 69 64 65 2a 28 69 64 78 2b 31 && Stride*(idx+1
07f0: 29 20 3c 3d 20 52 61 77 5f 52 20 29 0a 09 09 7b ) <= Raw_R )...{
0800: 0a 09 09 09 6d 65 20 5e 3d 20 31 3b 0a 09 09 09 ....me ^= 1;....
0810: 72 65 74 75 72 6e 3b 0a 09 09 7d 0a 0a 09 09 2f return;...}..../
0820: 2f 20 4f 76 65 72 6c 61 70 0a 09 09 74 72 65 65 / Overlap...tree
0830: 5b 6c 76 2d 31 5d 5b 69 64 78 2a 32 5d 2e 73 77 [lv-1][idx*2].sw
0840: 69 74 63 68 65 64 20 20 20 5e 3d 20 6d 65 3b 0a itched ^= me;.
0850: 09 09 74 72 65 65 5b 6c 76 2d 31 5d 5b 69 64 78 ..tree[lv-1][idx
0860: 2a 32 2b 31 5d 2e 73 77 69 74 63 68 65 64 20 5e *2+1].switched ^
0870: 3d 20 6d 65 3b 0a 09 09 53 77 69 74 63 68 52 65 = me;...SwitchRe
0880: 63 28 53 74 72 69 64 65 2f 32 2c 20 6c 76 2d 31 c(Stride/2, lv-1
0890: 2c 20 69 64 78 2a 32 2c 20 20 20 52 61 77 5f 4c , idx*2, Raw_L
08a0: 2c 20 52 61 77 5f 52 29 3b 0a 09 09 53 77 69 74 , Raw_R);...Swit
08b0: 63 68 52 65 63 28 53 74 72 69 64 65 2f 32 2c 20 chRec(Stride/2,
08c0: 6c 76 2d 31 2c 20 69 64 78 2a 32 2b 31 2c 20 52 lv-1, idx*2+1, R
08d0: 61 77 5f 4c 2c 20 52 61 77 5f 52 29 3b 0a 09 09 aw_L, Raw_R);...
08e0: 55 70 64 61 74 65 4d 69 64 4e 6f 64 65 28 6c 76 UpdateMidNode(lv
08f0: 2c 20 69 64 78 29 3b 0a 09 7d 0a 7d 3b 0a , idx);..}.};.