@@ -20,7 +20,7 @@ describe('FenwickTree', () => {
2020expect ( tree . treeArray . length ) . toBe ( inputArray . length + 1 ) ;
2121
2222inputArray . forEach ( ( value , index ) => {
23- tree . update ( index + 1 , value ) ;
23+ tree . increase ( index + 1 , value ) ;
2424} ) ;
2525
2626expect ( tree . treeArray ) . toEqual ( [ 0 , 3 , 5 , - 1 , 10 , 5 , 9 , - 3 , 19 , 7 , 9 , 3 ] ) ;
@@ -41,23 +41,42 @@ describe('FenwickTree', () => {
4141expect ( tree . queryRange ( 1 , 2 ) ) . toBe ( 5 ) ;
4242expect ( tree . queryRange ( 2 , 4 ) ) . toBe ( 7 ) ;
4343expect ( tree . queryRange ( 6 , 9 ) ) . toBe ( 11 ) ;
44+
45+ tree . increase ( 3 , 1 ) ;
46+
47+ expect ( tree . query ( 1 ) ) . toBe ( 3 ) ;
48+ expect ( tree . query ( 2 ) ) . toBe ( 5 ) ;
49+ expect ( tree . query ( 3 ) ) . toBe ( 5 ) ;
50+ expect ( tree . query ( 4 ) ) . toBe ( 11 ) ;
51+ expect ( tree . query ( 5 ) ) . toBe ( 16 ) ;
52+ expect ( tree . query ( 6 ) ) . toBe ( 20 ) ;
53+ expect ( tree . query ( 7 ) ) . toBe ( 17 ) ;
54+ expect ( tree . query ( 8 ) ) . toBe ( 20 ) ;
55+ expect ( tree . query ( 9 ) ) . toBe ( 27 ) ;
56+ expect ( tree . query ( 10 ) ) . toBe ( 29 ) ;
57+ expect ( tree . query ( 11 ) ) . toBe ( 32 ) ;
58+
59+ expect ( tree . queryRange ( 1 , 1 ) ) . toBe ( 3 ) ;
60+ expect ( tree . queryRange ( 1 , 2 ) ) . toBe ( 5 ) ;
61+ expect ( tree . queryRange ( 2 , 4 ) ) . toBe ( 8 ) ;
62+ expect ( tree . queryRange ( 6 , 9 ) ) . toBe ( 11 ) ;
4463} ) ;
4564
4665it ( 'should correctly execute queries' , ( ) => {
4766const tree = new FenwickTree ( 5 ) ;
4867
49- tree . update ( 1 , 4 ) ;
50- tree . update ( 3 , 7 ) ;
68+ tree . increase ( 1 , 4 ) ;
69+ tree . increase ( 3 , 7 ) ;
5170
5271expect ( tree . query ( 1 ) ) . toBe ( 4 ) ;
5372expect ( tree . query ( 3 ) ) . toBe ( 11 ) ;
5473expect ( tree . query ( 5 ) ) . toBe ( 11 ) ;
5574expect ( tree . queryRange ( 2 , 3 ) ) . toBe ( 7 ) ;
5675
57- tree . update ( 2 , 5 ) ;
76+ tree . increase ( 2 , 5 ) ;
5877expect ( tree . query ( 5 ) ) . toBe ( 16 ) ;
5978
60- tree . update ( 1 , 3 ) ;
79+ tree . increase ( 1 , 3 ) ;
6180expect ( tree . queryRange ( 1 , 1 ) ) . toBe ( 7 ) ;
6281expect ( tree . query ( 5 ) ) . toBe ( 19 ) ;
6382expect ( tree . queryRange ( 1 , 5 ) ) . toBe ( 19 ) ;
@@ -66,12 +85,12 @@ describe('FenwickTree', () => {
6685it ( 'should throw exceptions' , ( ) => {
6786const tree = new FenwickTree ( 5 ) ;
6887
69- const updateAtInvalidLowIndex = ( ) => {
70- tree . update ( 0 , 1 ) ;
88+ const increaseAtInvalidLowIndex = ( ) => {
89+ tree . increase ( 0 , 1 ) ;
7190} ;
7291
73- const updateAtInvalidHighIndex = ( ) => {
74- tree . update ( 10 , 1 ) ;
92+ const increaseAtInvalidHighIndex = ( ) => {
93+ tree . increase ( 10 , 1 ) ;
7594} ;
7695
7796const queryInvalidLowIndex = ( ) => {
@@ -86,8 +105,8 @@ describe('FenwickTree', () => {
86105tree . queryRange ( 3 , 2 ) ;
87106} ;
88107
89- expect ( updateAtInvalidLowIndex ) . toThrowError ( ) ;
90- expect ( updateAtInvalidHighIndex ) . toThrowError ( ) ;
108+ expect ( increaseAtInvalidLowIndex ) . toThrowError ( ) ;
109+ expect ( increaseAtInvalidHighIndex ) . toThrowError ( ) ;
91110expect ( queryInvalidLowIndex ) . toThrowError ( ) ;
92111expect ( queryInvalidHighIndex ) . toThrowError ( ) ;
93112expect ( rangeQueryInvalidIndex ) . toThrowError ( ) ;