整数数组区间求和问题的 Python 实现

以下是使用 Python 实现该功能的代码： ```python # 读取输入的 n 和 m n, m = map(int, input().split()) # 读取 a 数组 a = list(map(int, input().split())) # 读取 m 次询问的 k 值 queries = list(map(int, input().split())) # 遍历每次询问 for k in queries: # 计算从 a[k-1] 到 a[n-1] 的和 result = sum(a[k - 1:]) # 输出结果 print(result) ``` ### 代码解释 1. **读取输入**：首先读取两个整数 `n` 和 `m`，分别表示数组元素的数量和询问的次数。接着读取包含 `n` 个整数的数组 `a`，再读取包含 `m` 个整数的数组 `queries`，其中每个整数代表一次询问的 `k` 值。 2. **遍历询问**：对于 `queries` 中的每个 `k` 值，计算从 `a[k - 1]` 到 `a[n - 1]` 的元素之和。这里使用 Python 的 `sum` 函数对数组的切片进行求和。 3. **输出结果**：每次计算完成后，打印出求和的结果。 ### 复杂度分析 - **时间复杂度**：每次询问的时间复杂度为 $O(n - k + 1)$，在最坏情况下，每次询问的时间复杂度为 $O(n)$，总共进行 `m` 次询问，因此总的时间复杂度为 $O(m * n)$。 - **空间复杂度**：主要使用了数组 `a` 和 `queries`，因此空间复杂度为 $O(n + m)$。 ######[AI问答 | 403点数解答 | 2026-08-30 10:54:08]