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

以下是对代码的分析与修改。你给出的代码在处理大规模数据时，可能会因为整数溢出问题造成 `WA`（Wrong Answer）。特别是在计算 `(long long)a[i] * a[i]` 时，尽管使用了 `long long` 类型，但若 `a[i]` 数值过大，直接相乘仍有可能溢出。 下面是修改后的代码： ```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) { 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; } ``` ### 修改点说明： 在 `calculateTotalEnergy` 函数里，计算 `squareSum` 时，先对 `(long long)a[i] * a[i]` 取模再累加到 `squareSum` 中，防止乘法运算时溢出。 ######[AI问答 | 749点数解答 | 2026-05-31 16:09:42]