Ejemplo n.º 1
0
  @Test
  public void testFenwickTree() {
    nums = new int[] {3, 4, 9, 1, 4, 7, 6, 0};

    testClass = new FenwickTree(nums);
    testRangeSum();

    int val = +3;
    int updateIdx = 4;
    nums[updateIdx] += val;
    testClass.updateTree(updateIdx, val);
    testRangeSum();

    val = -6;
    updateIdx = 3;
    nums[updateIdx] += val;
    testClass.updateTree(updateIdx, val);
    testRangeSum();

    val = 7;
    updateIdx = 0;
    nums[updateIdx] -= val;
    testClass.updateTree(updateIdx, -val);

    val = -4;
    updateIdx = 1;
    nums[updateIdx] += val;
    testClass.updateTree(updateIdx, val);

    testRangeSum();
  }