Submission #23465233

---

Source Code Expand

import java.util.*;
import java.io.*;

public class Main{ 
  public static void main(String[] args) throws IOException {
    BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
    String[] nq = in.readLine().split(" ");
    long n = Long.parseLong(nq[0]);
    long q = Long.parseLong(nq[1]);
    String[] ast = in.readLine().split(" ");
    long stock = 0;
    List<Long[]> list = new ArrayList<>();

    long temp = 0;
    for (String as : ast) {
      long a = Long.parseLong(as);
      if (temp != a-stock) {
        list.add(new Long[]{temp, stock});
        temp = a-stock;
      }
      stock++;
    }
    list.add(new Long[]{temp,stock});
    list.add(new Long[]{Long.MAX_VALUE,Long.MAX_VALUE});

    StringBuilder sb = new StringBuilder();
    for (int i = 0; i < q; i++) {
      long query = Long.parseLong(in.readLine());
      int min = 0; 
      int max = list.size()-1;
      int curr = (min+max)/2;
      while (min+1 != max) {
        if (list.get(curr)[0] > query) {
          max = curr;
        } else {
          min = curr;
        }
        curr = (min+max)/2;
      }
      sb.append(query + list.get(min)[1]);
      sb.append('\n');
    }
    System.out.println(sb);
  }
}

Submission Info

Judge Result