コード例 #1
0
ファイル: tree_test.py プロジェクト: nulpunkt/segment-tree
 def test_construct_mini_tree(self):
     tree = SegmentTree([1,2,3,4])
     self.assertEquals(10, tree.sum(0, 3))
コード例 #2
0
ファイル: tree_test.py プロジェクト: nulpunkt/segment-tree
 def test_construct_terminal(self):
     tree = SegmentTree([1,2])
     self.assertEquals(3, tree.sum(0, 1))
コード例 #3
0
ファイル: tree_test.py プロジェクト: nulpunkt/segment-tree
 def test_set_leaf(self):
     tree = SegmentTree([1,2,3,4])
     self.assertEquals(2, tree.sum(1, 1))
     tree.set(1, 10)
     self.assertEquals(10, tree.sum(1, 1))
コード例 #4
0
ファイル: tree_test.py プロジェクト: nulpunkt/segment-tree
 def test_set_internal(self):
     tree = SegmentTree([1,2,3,4])
     self.assertEquals(10, tree.sum(0, 3))
     tree.set(1, 10)
     self.assertEquals(11, tree.sum(0, 1))
     self.assertEquals(18, tree.sum(0, 3))
コード例 #5
0
ファイル: tree_test.py プロジェクト: nulpunkt/segment-tree
 def test_partial_sum(self):
     tree = SegmentTree([1,2,3,4,12,13])
     self.assertEquals(3, tree.sum(0, 1))
     self.assertEquals(7, tree.sum(2, 3))
     self.assertEquals(21, tree.sum(1, 4))