C++使用前缀和后缀和寻找数组平衡点下标

### 🍃思路分析 本题要求找到数组中的平衡点下标，使得平衡点左侧的数之和等于平衡点右侧的数之和。为了高效地解决这个问题，我们可以使用前缀和与后缀和的方法。前缀和数组 `prefixSum` 用于存储从数组起始位置到当前位置的元素之和，后缀和数组 `suffixSum` 用于存储从数组末尾位置到当前位置的元素之和。通过遍历数组，比较每个位置的前缀和与后缀和，找到满足条件的平衡点下标。 ### 💻代码实现 ```cpp #include <iostream> #include <vector> using namespace std; int findBalancePoint(int n, vector<int>& a) { // 前缀和数组 vector<int> prefixSum(n + 1, 0); // 后缀和数组 vector<int> suffixSum(n + 2, 0); // 计算前缀和 for (int i = 1; i <= n; ++i) { prefixSum[i] = prefixSum[i - 1] + a[i - 1]; } // 计算后缀和 for (int i = n; i >= 1; --i) { suffixSum[i] = suffixSum[i + 1] + a[i - 1]; } // 寻找平衡点 for (int i = 1; i <= n; ++i) { if (prefixSum[i - 1] == suffixSum[i + 1]) { return i; } } return -1; // 如果没有找到平衡点，返回 -1 } int main() { int n; cin >> n; vector<int> a(n); for (int i = 0; i < n; ++i) { cin >> a[i]; } int balancePoint = findBalancePoint(n, a); cout << balancePoint << endl; return 0; } ``` ### 📝代码解释 1. **前缀和数组 `prefixSum`**：用于存储从数组起始位置到当前位置的元素之和。`prefixSum[i]` 表示前 `i` 个元素的和。 2. **后缀和数组 `suffixSum`**：用于存储从数组末尾位置到当前位置的元素之和。`suffixSum[i]` 表示从第 `i` 个元素到数组末尾的元素之和。 3. **寻找平衡点**：遍历数组，对于每个位置 `i`，比较 `prefixSum[i - 1]` 和 `suffixSum[i + 1]`。如果两者相等，则 `i` 就是平衡点的下标。 ### ⚙️复杂度分析 - **时间复杂度**：$O(n)$，其中 $n$ 是数组的长度。需要遍历数组两次来计算前缀和和后缀和，然后再遍历一次数组来寻找平衡点。 - **空间复杂度**：$O(n)$，主要用于存储前缀和和后缀和数组。 ### 🎯开发建议 - **边界条件处理**：在实际开发中，需要考虑数组为空或只有一个元素的情况，确保代码的健壮性。 - **代码复用**：将计算前缀和、后缀和以及寻找平衡点的逻辑封装成独立的函数，提高代码的复用性。 - **错误处理**：如果没有找到平衡点，返回一个特殊值（如 -1），并在调用函数时进行相应的错误处理。 ######[AI写代码神器 | 789点数解答 | 2026-04-04 20:06:22]