kyber_test_idx

kyber_test_idx.go

func check(j, arrLength int) bool {
  return j < (arrLength-1)
}

func Query1(arr []int, queries [][2]int) {
  var queryResult int
  for _, q := range queries {
    if check(q[1], len(arr)) {
      log.Fatalf("Invalid query: %d, %d", q[0], q[1])
    }

    for i := q[0]; i <= q[1]; i++ {
      queryResult += arr[i]
    }

    fmt.Printf("Query %d, %d result %d", q[0], q[1], queryResult)
    queryResult = 0
  }
}

func Query2(arr []int, query [][2]int) {
  if len(arr) < 1 {
    log.Fatalf("empty input array")
  }  

  var sumArr = make([]int, len(arr))
  sumArr[0] = arr[0]
  for i := 1; i < len(arr); i++ {
    sumArr[i] = sumArr[i-1] + arr[i]
  }

  for _, q := range queries {
    if check(q[1], len(arr)) {
      log.Fatalf("Invalid query: %d, %d", q[0], q[1])
    }
    fmt.Printf("Query %d, %d result %d", q[0], q[1], sumArr[j] - sumArr[i] + arr[i])
  }
}