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 32 30 - Codeforces 20
0090: 30 20 44 0a 2f 2f 20 20 20 2d 20 43 6f 64 65 66 0 D.// - Codef
00a0: 6f 72 63 65 73 20 31 30 37 20 43 20 28 6f 6c 64 orces 107 C (old
00b0: 20 76 65 72 73 69 6f 6e 29 0a 2f 2f 20 20 20 2d version).// -
00c0: 20 43 6f 64 65 66 6f 72 63 65 73 20 31 30 34 20 Codeforces 104
00d0: 45 20 28 6f 6c 64 20 76 65 72 73 69 6f 6e 29 0a E (old version).
00e0: 2f 2f 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d //--------------
00f0: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
0100: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d ----------------
0110: 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 2d 0a ---------------.
0120: 0a 63 6c 61 73 73 20 53 65 67 6d 65 6e 74 54 72 .class SegmentTr
0130: 65 65 0a 7b 0a 09 73 74 72 75 63 74 20 4e 6f 64 ee.{..struct Nod
0140: 65 0a 09 7b 0a 09 09 69 6e 74 20 73 75 6d 3b 0a e..{...int sum;.
0150: 0a 09 09 73 74 61 74 69 63 20 4e 6f 64 65 20 5a ...static Node Z
0160: 65 72 6f 28 29 0a 09 09 7b 0a 09 09 09 4e 6f 64 ero()...{....Nod
0170: 65 20 63 20 3d 20 7b 30 7d 3b 0a 09 09 09 72 65 e c = {0};....re
0180: 74 75 72 6e 20 63 3b 0a 09 09 7d 0a 09 09 73 74 turn c;...}...st
0190: 61 74 69 63 20 4e 6f 64 65 20 4f 6e 65 28 69 6e atic Node One(in
01a0: 74 20 76 29 0a 09 09 7b 0a 09 09 09 4e 6f 64 65 t v)...{....Node
01b0: 20 63 20 3d 20 7b 76 7d 3b 0a 09 09 09 72 65 74 c = {v};....ret
01c0: 75 72 6e 20 63 3b 0a 09 09 7d 0a 09 09 73 74 61 urn c;...}...sta
01d0: 74 69 63 20 4e 6f 64 65 20 43 6f 6e 63 61 74 28 tic Node Concat(
01e0: 63 6f 6e 73 74 20 4e 6f 64 65 26 20 6c 2c 20 63 const Node& l, c
01f0: 6f 6e 73 74 20 4e 6f 64 65 26 20 72 29 0a 09 09 onst Node& r)...
0200: 7b 0a 09 09 09 4e 6f 64 65 20 63 20 3d 20 7b 6c {....Node c = {l
0210: 2e 73 75 6d 20 2b 20 72 2e 73 75 6d 7d 3b 0a 09 .sum + r.sum};..
0220: 09 09 72 65 74 75 72 6e 20 63 3b 0a 09 09 7d 0a ..return c;...}.
0230: 09 09 73 74 61 74 69 63 20 4e 6f 64 65 20 52 65 ..static Node Re
0240: 70 65 61 74 28 63 6f 6e 73 74 20 4e 6f 64 65 26 peat(const Node&
0250: 20 6e 2c 20 69 6e 74 20 6b 29 0a 09 09 7b 0a 09 n, int k)...{..
0260: 09 09 69 66 28 6b 3d 3d 30 29 20 72 65 74 75 72 ..if(k==0) retur
0270: 6e 20 5a 65 72 6f 28 29 3b 0a 09 09 09 4e 6f 64 n Zero();....Nod
0280: 65 20 63 20 3d 20 7b 6e 2e 73 75 6d 20 2a 20 6b e c = {n.sum * k
0290: 7d 3b 0a 09 09 09 72 65 74 75 72 6e 20 63 3b 0a };....return c;.
02a0: 09 09 7d 0a 0a 09 09 62 6f 6f 6c 20 6c 61 7a 79 ..}....bool lazy
02b0: 3b 0a 09 7d 3b 0a 0a 70 75 62 6c 69 63 3a 0a 09 ;..};..public:..
02c0: 74 65 6d 70 6c 61 74 65 3c 74 79 70 65 6e 61 6d template<typenam
02d0: 65 20 53 65 71 3e 0a 09 53 65 67 6d 65 6e 74 54 e Seq>..SegmentT
02e0: 72 65 65 28 63 6f 6e 73 74 20 53 65 71 26 20 73 ree(const Seq& s
02f0: 29 20 7b 0a 09 09 69 6e 74 20 4e 20 3d 20 31 3b ) {...int N = 1;
0300: 0a 09 09 77 68 69 6c 65 28 20 4e 20 3c 20 73 2e ...while( N < s.
0310: 73 69 7a 65 28 29 20 29 0a 09 09 09 4e 20 3c 3c size() )....N <<
0320: 3d 20 31 3b 0a 0a 09 09 74 72 65 65 2e 72 65 73 = 1;....tree.res
0330: 69 7a 65 28 74 72 65 65 2e 73 69 7a 65 28 29 2b ize(tree.size()+
0340: 31 29 3b 20 74 72 65 65 2e 62 61 63 6b 28 29 2e 1); tree.back().
0350: 72 65 73 69 7a 65 28 4e 29 3b 0a 09 09 66 6f 72 resize(N);...for
0360: 28 69 6e 74 20 69 3d 30 3b 20 69 3c 4e 3b 20 2b (int i=0; i<N; +
0370: 2b 69 29 0a 09 09 09 74 72 65 65 2e 62 61 63 6b +i)....tree.back
0380: 28 29 5b 69 5d 20 3d 20 69 3c 73 2e 73 69 7a 65 ()[i] = i<s.size
0390: 28 29 20 3f 20 4e 6f 64 65 3a 3a 4f 6e 65 28 73 () ? Node::One(s
03a0: 5b 69 5d 29 20 3a 20 4e 6f 64 65 3a 3a 5a 65 72 [i]) : Node::Zer
03b0: 6f 28 29 3b 0a 0a 09 09 77 68 69 6c 65 28 4e 3e o();....while(N>
03c0: 3e 3d 31 29 20 7b 0a 09 09 09 74 72 65 65 2e 72 >=1) {....tree.r
03d0: 65 73 69 7a 65 28 74 72 65 65 2e 73 69 7a 65 28 esize(tree.size(
03e0: 29 2b 31 29 3b 20 74 72 65 65 2e 62 61 63 6b 28 )+1); tree.back(
03f0: 29 2e 72 65 73 69 7a 65 28 4e 29 3b 0a 09 09 09 ).resize(N);....
0400: 66 6f 72 28 69 6e 74 20 69 3d 30 3b 20 69 3c 4e for(int i=0; i<N
0410: 3b 20 2b 2b 69 29 0a 09 09 09 09 43 61 6c 63 4d ; ++i).....CalcM
0420: 69 64 4e 6f 64 65 28 74 72 65 65 2e 73 69 7a 65 idNode(tree.size
0430: 28 29 2d 31 2c 20 69 29 3b 0a 09 09 7d 0a 09 7d ()-1, i);...}..}
0440: 0a 0a 09 4e 6f 64 65 20 51 75 65 72 79 28 69 6e ...Node Query(in
0450: 74 20 73 2c 20 69 6e 74 20 65 29 20 7b 20 2f 2f t s, int e) { //
0460: 20 63 6f 6d 70 75 74 65 20 68 28 20 73 65 71 5b compute h( seq[
0470: 73 2c 65 29 20 29 20 3a 20 20 4f 28 6c 6f 67 20 s,e) ) : O(log
0480: 6e 29 0a 09 09 72 65 74 75 72 6e 20 51 75 65 72 n)...return Quer
0490: 79 52 65 63 28 73 2c 20 65 2c 20 74 72 65 65 2e yRec(s, e, tree.
04a0: 73 69 7a 65 28 29 2d 31 2c 20 30 2c 20 74 72 65 size()-1, 0, tre
04b0: 65 5b 30 5d 2e 73 69 7a 65 28 29 29 3b 0a 09 7d e[0].size());..}
04c0: 0a 0a 09 74 65 6d 70 6c 61 74 65 3c 74 79 70 65 ...template<type
04d0: 6e 61 6d 65 20 56 61 6c 75 65 3e 0a 09 76 6f 69 name Value>..voi
04e0: 64 20 53 65 74 28 69 6e 74 20 73 2c 20 69 6e 74 d Set(int s, int
04f0: 20 65 2c 20 56 61 6c 75 65 20 76 29 20 7b 20 2f e, Value v) { /
0500: 2f 20 73 65 71 5b 73 2c 65 29 3a 3d 76 20 3a 20 / seq[s,e):=v :
0510: 4f 28 6c 6f 67 20 6e 29 0a 09 09 53 65 74 52 65 O(log n)...SetRe
0520: 63 28 73 2c 20 65 2c 20 4e 6f 64 65 3a 3a 4f 6e c(s, e, Node::On
0530: 65 28 76 29 2c 20 74 72 65 65 2e 73 69 7a 65 28 e(v), tree.size(
0540: 29 2d 31 2c 20 30 2c 20 74 72 65 65 5b 30 5d 2e )-1, 0, tree[0].
0550: 73 69 7a 65 28 29 29 3b 0a 09 7d 0a 0a 70 72 69 size());..}..pri
0560: 76 61 74 65 3a 0a 09 4e 6f 64 65 20 51 75 65 72 vate:..Node Quer
0570: 79 52 65 63 28 69 6e 74 20 73 2c 20 69 6e 74 20 yRec(int s, int
0580: 65 2c 20 69 6e 74 20 6c 76 2c 20 69 6e 74 20 69 e, int lv, int i
0590: 64 78 2c 20 69 6e 74 20 73 74 72 69 64 65 29 0a dx, int stride).
05a0: 09 7b 0a 09 09 63 6f 6e 73 74 20 69 6e 74 20 6d .{...const int m
05b0: 79 4c 20 3d 20 73 74 72 69 64 65 2a 69 64 78 3b yL = stride*idx;
05c0: 0a 09 09 63 6f 6e 73 74 20 69 6e 74 20 6d 79 52 ...const int myR
05d0: 20 3d 20 73 74 72 69 64 65 2a 28 69 64 78 2b 31 = stride*(idx+1
05e0: 29 3b 0a 09 09 69 66 28 20 65 3c 3d 6d 79 4c 20 );...if( e<=myL
05f0: 7c 7c 20 6d 79 52 3c 3d 73 20 29 0a 09 09 09 72 || myR<=s )....r
0600: 65 74 75 72 6e 20 4e 6f 64 65 3a 3a 5a 65 72 6f eturn Node::Zero
0610: 28 29 3b 0a 09 09 52 65 73 6f 6c 76 65 4c 61 7a ();...ResolveLaz
0620: 79 28 6c 76 2c 20 69 64 78 29 3b 0a 0a 09 09 69 y(lv, idx);....i
0630: 66 28 20 73 3c 3d 6d 79 4c 20 26 26 20 6d 79 52 f( s<=myL && myR
0640: 3c 3d 65 20 29 0a 09 09 09 72 65 74 75 72 6e 20 <=e )....return
0650: 74 72 65 65 5b 6c 76 5d 5b 69 64 78 5d 3b 0a 09 tree[lv][idx];..
0660: 09 72 65 74 75 72 6e 20 4e 6f 64 65 3a 3a 43 6f .return Node::Co
0670: 6e 63 61 74 28 51 75 65 72 79 52 65 63 28 73 2c ncat(QueryRec(s,
0680: 65 2c 6c 76 2d 31 2c 69 64 78 2a 32 2c 73 74 72 e,lv-1,idx*2,str
0690: 69 64 65 2f 32 29 2c 0a 09 09 20 20 20 20 20 20 ide/2),...
06a0: 20 20 20 20 20 20 20 20 20 20 20 20 20 20 51 75 Qu
06b0: 65 72 79 52 65 63 28 73 2c 65 2c 6c 76 2d 31 2c eryRec(s,e,lv-1,
06c0: 69 64 78 2a 32 2b 31 2c 73 74 72 69 64 65 2f 32 idx*2+1,stride/2
06d0: 29 29 3b 0a 09 7d 0a 0a 09 76 6f 69 64 20 53 65 ));..}...void Se
06e0: 74 52 65 63 28 69 6e 74 20 73 2c 20 69 6e 74 20 tRec(int s, int
06f0: 65 2c 20 63 6f 6e 73 74 20 4e 6f 64 65 26 20 6e e, const Node& n
0700: 2c 20 69 6e 74 20 6c 76 2c 20 69 6e 74 20 69 64 , int lv, int id
0710: 78 2c 20 69 6e 74 20 73 74 72 69 64 65 29 0a 09 x, int stride)..
0720: 7b 0a 09 09 63 6f 6e 73 74 20 69 6e 74 20 6d 79 {...const int my
0730: 4c 20 3d 20 73 74 72 69 64 65 2a 69 64 78 3b 0a L = stride*idx;.
0740: 09 09 63 6f 6e 73 74 20 69 6e 74 20 6d 79 52 20 ..const int myR
0750: 3d 20 73 74 72 69 64 65 2a 28 69 64 78 2b 31 29 = stride*(idx+1)
0760: 3b 0a 09 09 69 66 28 20 65 3c 3d 6d 79 4c 20 7c ;...if( e<=myL |
0770: 7c 20 6d 79 52 3c 3d 73 20 29 0a 09 09 09 72 65 | myR<=s )....re
0780: 74 75 72 6e 3b 0a 09 09 52 65 73 6f 6c 76 65 4c turn;...ResolveL
0790: 61 7a 79 28 6c 76 2c 20 69 64 78 29 3b 0a 0a 09 azy(lv, idx);...
07a0: 09 69 66 28 20 73 74 72 69 64 65 20 3d 3d 20 31 .if( stride == 1
07b0: 20 29 20 7b 0a 09 09 09 74 72 65 65 5b 6c 76 5d ) {....tree[lv]
07c0: 5b 69 64 78 5d 20 3d 20 6e 3b 0a 09 09 7d 20 65 [idx] = n;...} e
07d0: 6c 73 65 20 7b 0a 09 09 09 69 66 28 20 73 3c 3d lse {....if( s<=
07e0: 6d 79 4c 20 26 26 20 6d 79 52 3c 3d 65 20 29 20 myL && myR<=e )
07f0: 7b 0a 09 09 09 09 74 72 65 65 5b 6c 76 5d 5b 69 {.....tree[lv][i
0800: 64 78 5d 20 3d 20 6e 3b 0a 09 09 09 09 74 72 65 dx] = n;.....tre
0810: 65 5b 6c 76 5d 5b 69 64 78 5d 2e 6c 61 7a 79 20 e[lv][idx].lazy
0820: 3d 20 74 72 75 65 3b 0a 09 09 09 09 52 65 73 6f = true;.....Reso
0830: 6c 76 65 4c 61 7a 79 28 6c 76 2c 20 69 64 78 29 lveLazy(lv, idx)
0840: 3b 0a 09 09 09 09 72 65 74 75 72 6e 3b 0a 09 09 ;.....return;...
0850: 09 7d 0a 09 09 09 53 65 74 52 65 63 28 73 2c 65 .}....SetRec(s,e
0860: 2c 6e 2c 6c 76 2d 31 2c 69 64 78 2a 32 2c 73 74 ,n,lv-1,idx*2,st
0870: 72 69 64 65 2f 32 29 3b 0a 09 09 09 53 65 74 52 ride/2);....SetR
0880: 65 63 28 73 2c 65 2c 6e 2c 6c 76 2d 31 2c 69 64 ec(s,e,n,lv-1,id
0890: 78 2a 32 2b 31 2c 73 74 72 69 64 65 2f 32 29 3b x*2+1,stride/2);
08a0: 0a 09 09 09 43 61 6c 63 4d 69 64 4e 6f 64 65 28 ....CalcMidNode(
08b0: 6c 76 2c 20 69 64 78 29 3b 0a 09 09 7d 0a 09 7d lv, idx);...}..}
08c0: 0a 0a 09 76 6f 69 64 20 43 61 6c 63 4d 69 64 4e ...void CalcMidN
08d0: 6f 64 65 28 69 6e 74 20 6c 76 2c 20 69 6e 74 20 ode(int lv, int
08e0: 69 64 78 29 0a 09 7b 0a 09 09 74 72 65 65 5b 6c idx)..{...tree[l
08f0: 76 5d 5b 69 64 78 5d 20 3d 20 4e 6f 64 65 3a 3a v][idx] = Node::
0900: 43 6f 6e 63 61 74 28 74 72 65 65 5b 6c 76 2d 31 Concat(tree[lv-1
0910: 5d 5b 69 64 78 2a 32 5d 2c 20 74 72 65 65 5b 6c ][idx*2], tree[l
0920: 76 2d 31 5d 5b 69 64 78 2a 32 2b 31 5d 29 3b 0a v-1][idx*2+1]);.
0930: 09 7d 0a 0a 09 76 6f 69 64 20 52 65 73 6f 6c 76 .}...void Resolv
0940: 65 4c 61 7a 79 28 69 6e 74 20 6c 76 2c 20 69 6e eLazy(int lv, in
0950: 74 20 69 64 78 29 0a 09 7b 0a 09 09 69 66 28 74 t idx)..{...if(t
0960: 72 65 65 5b 6c 76 5d 5b 69 64 78 5d 2e 6c 61 7a ree[lv][idx].laz
0970: 79 29 20 7b 0a 09 09 09 69 66 28 6c 76 20 3e 20 y) {....if(lv >
0980: 30 29 20 7b 0a 09 09 09 09 74 72 65 65 5b 6c 76 0) {.....tree[lv
0990: 2d 31 5d 5b 69 64 78 2a 32 5d 20 20 20 3d 20 74 -1][idx*2] = t
09a0: 72 65 65 5b 6c 76 5d 5b 69 64 78 5d 3b 0a 09 09 ree[lv][idx];...
09b0: 09 09 74 72 65 65 5b 6c 76 2d 31 5d 5b 69 64 78 ..tree[lv-1][idx
09c0: 2a 32 2b 31 5d 20 3d 20 74 72 65 65 5b 6c 76 5d *2+1] = tree[lv]
09d0: 5b 69 64 78 5d 3b 0a 09 09 09 7d 0a 09 09 09 74 [idx];....}....t
09e0: 72 65 65 5b 6c 76 5d 5b 69 64 78 5d 20 3d 20 4e ree[lv][idx] = N
09f0: 6f 64 65 3a 3a 52 65 70 65 61 74 28 74 72 65 65 ode::Repeat(tree
0a00: 5b 6c 76 5d 5b 69 64 78 5d 2c 20 31 3c 3c 6c 76 [lv][idx], 1<<lv
0a10: 29 3b 0a 09 09 7d 0a 09 7d 0a 0a 09 76 65 63 74 );...}..}...vect
0a20: 6f 72 3c 20 76 65 63 74 6f 72 3c 4e 6f 64 65 3e or< vector<Node>
0a30: 20 3e 20 74 72 65 65 3b 0a 7d 3b 0a > tree;.};.