解决代码WA问题：能量波汇合新能量总和计算代码修正

### 🐞问题分析 代码出现 `WA`（Wrong Answer）错误，可能是以下几个方面出了问题： 1. **输入验证**：虽然对 `n` 进行了是否小于等于 0 的检查，但没有对 `a[i]` 的输入范围进行检查。 2. **数据类型**：在计算过程中，虽然使用了 `long long` 来避免部分溢出问题，但仍可能存在一些边界情况导致溢出。 3. **模运算**：在模运算过程中，如果中间结果出现负数，可能会导致结果错误。 4. **逆元计算**：扩展欧几里得算法求逆元部分可能存在错误。 ### 💡解决方案 以下是修正后的代码： ```cpp #include <iostream> const int MOD = 1000000007; // 扩展欧几里得算法求逆元 int modInverse(int a, int m) { int m0 = m, t, q; int x0 = 0, x1 = 1; if (m == 1) return 0; while (a > 1) { // q 是商 q = a / m; t = m; // m 是余数 m = a % m, a = t; t = x0; x0 = x1 - q * x0; x1 = t; } // 确保结果为正 if (x1 < 0) x1 += m0; return x1; } // 计算汇合后的新能量总和 long long calculateTotalEnergy(int n, int a[]) { long long sum = 0; // 存储所有能量波强度的和 long long squareSum = 0; // 存储所有能量波强度平方的和 // 计算所有能量波强度的和以及平方和 for (int i = 0; i < n; ++i) { // 确保输入的 a[i] 是合法的 if (a[i] < 0) { return 0; } sum = (sum + a[i]) % MOD; // 避免乘法溢出 squareSum = (squareSum + (long long)a[i] * a[i] % MOD) % MOD; } // 根据公式计算结果 long long result = (sum * sum % MOD - squareSum + MOD) % MOD; // 因为要除以2，而在模运算中不能直接除，需要乘以2的模逆元 int inv2 = modInverse(2, MOD); result = result * inv2 % MOD; return result; } int main() { int n; std::cin >> n; // 检查输入的 n 是否合法 if (n <= 0) { std::cout << 0 << std::endl; return 0; } int *a = new int[n]; // 读取每个能量波的强度 for (int i = 0; i < n; ++i) { std::cin >> a[i]; } // 计算并输出汇合后的新能量总和 std::cout << calculateTotalEnergy(n, a) << std::endl; delete[] a; return 0; } ``` ### 📝代码解释 1. **输入验证**：在 `calculateTotalEnergy` 函数中，增加了对 `a[i]` 是否小于 0 的检查，如果小于 0 则直接返回 0。 2. **数据类型**：使用 `long long` 类型来存储中间结果，避免乘法溢出。 3. **模运算**：在计算 `result` 时，先加上 `MOD` 再取模，确保结果为正。 4. **逆元计算**：扩展欧几里得算法求逆元部分保持不变。 ######[AI问答 | 926点数解答 | 2026-05-31 16:16:12]