結果
問題 | No.1890 Many Sequences Sum Queries |
ユーザー | tenten |
提出日時 | 2022-04-06 10:24:56 |
言語 | Java21 (openjdk 21) |
結果 |
AC
|
実行時間 | 321 ms / 2,000 ms |
コード長 | 2,035 bytes |
コンパイル時間 | 2,149 ms |
コンパイル使用メモリ | 78,468 KB |
実行使用メモリ | 50,684 KB |
最終ジャッジ日時 | 2024-05-05 17:56:00 |
合計ジャッジ時間 | 6,508 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 321 ms
49,860 KB |
testcase_01 | AC | 266 ms
48,768 KB |
testcase_02 | AC | 45 ms
37,092 KB |
testcase_03 | AC | 45 ms
36,660 KB |
testcase_04 | AC | 44 ms
37,084 KB |
testcase_05 | AC | 45 ms
37,084 KB |
testcase_06 | AC | 46 ms
37,080 KB |
testcase_07 | AC | 216 ms
47,660 KB |
testcase_08 | AC | 135 ms
40,928 KB |
testcase_09 | AC | 133 ms
41,540 KB |
testcase_10 | AC | 259 ms
48,168 KB |
testcase_11 | AC | 201 ms
43,780 KB |
testcase_12 | AC | 290 ms
50,340 KB |
testcase_13 | AC | 291 ms
50,684 KB |
testcase_14 | AC | 215 ms
47,372 KB |
testcase_15 | AC | 45 ms
36,848 KB |
testcase_16 | AC | 45 ms
36,480 KB |
testcase_17 | AC | 47 ms
37,072 KB |
testcase_18 | AC | 45 ms
36,904 KB |
testcase_19 | AC | 46 ms
37,084 KB |
testcase_20 | AC | 47 ms
37,232 KB |
testcase_21 | AC | 44 ms
36,608 KB |
testcase_22 | AC | 44 ms
37,244 KB |
testcase_23 | AC | 45 ms
37,148 KB |
testcase_24 | AC | 46 ms
36,848 KB |
testcase_25 | AC | 47 ms
37,024 KB |
ソースコード
import java.io.*; import java.util.*; public class Main { public static void main(String[] args) throws Exception { Scanner sc = new Scanner(); int n = sc.nextInt(); int q = sc.nextInt(); int[] values = new int[n]; int[] counts = new int[n + 1]; TreeMap<Long, Integer> map = new TreeMap<>(); long prev = 0; for (int i = 0; i < n; i++) { map.put(prev, i); values[i] = sc.nextInt(); prev += (long)values[i] * (values[i] + 1) / 2; counts[i + 1] = counts[i] + values[i]; } StringBuilder sb = new StringBuilder(); for (int i = 0; i < q; i++) { long s = sc.nextInt(); if (prev < s) { sb.append("-1\n"); continue; } long current = map.lowerKey(s); int idx = map.get(current); int x = counts[idx]; long left = 0; long right = values[idx]; while (right - left > 1) { long m = (left + right) / 2; if (m * (m + 1) / 2 + current < s) { left = m; } else { right = m; } } sb.append(x + right).append("\n"); } System.out.print(sb); } } class Scanner { BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); StringTokenizer st = new StringTokenizer(""); public Scanner() throws Exception { } public int nextInt() throws Exception { return Integer.parseInt(next()); } public long nextLong() throws Exception { return Long.parseLong(next()); } public double nextDouble() throws Exception { return Double.parseDouble(next()); } public String next() throws Exception { if (!st.hasMoreTokens()) { st = new StringTokenizer(br.readLine()); } return st.nextToken(); } }