1. 题目背景与需求分析
P2671 [NOIP 2015 普及组] 求和是一道典型的信奥赛模拟题，考察选手对基础算法和数据结构的掌握程度。题目要求我们处理一个包含n个整数的序列，找出所有满足特定条件的元素对，并计算这些元素对的乘积之和。
这道题在当年比赛中属于中等难度，主要考察以下几个核心能力：
- 对题目条件的准确理解和转化能力
- 基础数据结构的灵活运用
- 时间复杂度优化的意识
- 边界条件的处理能力
在实际解题过程中，我发现很多初学者容易陷入以下几个误区：
- 直接使用暴力双重循环导致超时
- 没有充分理解题目中的数学规律
- 忽略了数据范围导致的整数溢出问题
- 对特殊情况的处理不完善
2. 题目详细解析与数学建模
2.1 题目条件拆解
题目给出的核心条件是：对于序列中的元素x和y，当满足以下三个条件时需要计入统计：
- x和y的颜色相同
- x和y的编号同为奇数或同为偶数
- x和y的编号满足x < y
这三个条件看似简单，但组合起来会产生一些有趣的数学性质。我们需要将其转化为可计算的数学表达式。
2.2 数学规律发现
通过分析我们可以发现，对于同一颜色且编号奇偶性相同的元素，它们的乘积在最终求和时会出现特定的组合规律。具体来说：
假设某颜色且同奇偶的元素的值为a1, a2, ..., ak，那么这些元素之间的乘积和为：
a1a2 + a1a3 + ... + a1ak + a2a3 + ... + a(k-1)*ak
这个表达式可以通过数学变换简化为：
[(a1 + a2 + ... + ak)^2 - (a1^2 + a2^2 + ... + ak^2)] / 2
这个发现是解题的关键，它将O(n^2)的时间复杂度降低到了O(n)。
3. C++实现方案
3.1 数据结构设计
基于上述分析，我们需要设计合适的数据结构来存储和计算相关信息：
cpp复制#include <iostream>
#include <vector>
#include <unordered_map>
using namespace std;
struct Group {
long long sum = 0; // 元素和
long long square_sum = 0; // 元素平方和
int count = 0; // 元素个数
};
int main() {
int n, m;
cin >> n >> m;
vector<int> color(n+1); // 颜色数组，索引从1开始
vector<int> number(n+1); // 数值数组
// 使用嵌套的unordered_map存储分组信息
// 外层key是颜色，内层key是奇偶性(0偶1奇)
unordered_map<int, unordered_map<int, Group>> groups;
// 读取输入数据
for(int i = 1; i <= n; i++) {
cin >> number[i];
}
for(int i = 1; i <= n; i++) {
cin >> color[i];
int parity = i % 2; // 奇偶性
int num = number[i];
// 更新对应分组的统计信息
groups[color[i]][parity].sum += num;
groups[color[i]][parity].square_sum += num * num;
groups[color[i]][parity].count++;
}
// 计算结果
long long result = 0;
const int MOD = 10007;
for(const auto& color_group : groups) {
for(const auto& parity_group : color_group.second) {
const Group& g = parity_group.second;
if(g.count >= 2) {
long long temp = (g.sum * g.sum - g.square_sum) / 2;
result = (result + temp) % MOD;
}
}
}
cout << result << endl;
return 0;
}
3.2 关键代码解析
-
数据结构选择：
- 使用
unordered_map嵌套结构来按颜色和奇偶性分组 - 每个分组维护sum、square_sum和count三个统计量
- 使用
-
输入处理：
- 分别读取数值和颜色信息
- 根据颜色和位置奇偶性将元素分配到不同组
-
计算结果：
- 应用推导出的数学公式计算每组的贡献
- 使用模运算防止溢出
-
时间复杂度分析：
- 输入处理：O(n)
- 结果计算：O(m)（m为颜色种类数）
- 总体复杂度：O(n)，完全满足题目要求
4. 优化技巧与注意事项
4.1 常见错误与修正
在实际编码过程中，容易出现以下问题：
-
整数溢出：
- 未使用long long导致中间结果溢出
- 修正：所有累加变量使用long long
-
模运算时机不当：
- 在除法前取模会导致计算结果错误
- 修正：先完成所有运算再取模
-
边界条件处理：
- 忽略count<2的情况
- 修正：添加条件判断
4.2 性能优化建议
-
IO优化：
cpp复制ios::sync_with_stdio(false); cin.tie(0);对于大规模输入，关闭同步可以显著提高读取速度
-
内存优化：
如果颜色范围已知且较小，可以用数组代替unordered_map -
并行计算：
对于多组测试数据的情况，可以考虑并行处理不同颜色组
5. 测试用例设计与验证
5.1 典型测试用例
plaintext复制输入1：
6 3
1 2 3 4 5 6
1 2 3 1 2 3
输出1：
42
输入2：
10 5
1 2 3 4 5 6 7 8 9 10
1 2 3 4 5 1 2 3 4 5
输出2：
440
5.2 边界测试用例
plaintext复制输入3：
2 1
10000 10000
1 1
输出3：
100000000 % 10007 = 9993
输入4：
1 1
12345
1
输出4：
0 (不足两个元素)
6. 算法扩展与变种思考
6.1 类似题目推荐
-
连续子序列求和：
- 求满足特定条件的连续子序列的和
- 可以使用前缀和优化
-
三元组计数：
- 扩展为三个元素的条件组合
- 需要更复杂的数学推导
-
动态版本：
- 支持中途修改元素值
- 需要引入线段树等数据结构
6.2 高阶优化思路
对于更大规模的数据，可以考虑以下优化：
-
分块处理：
将数据分成若干块，分别处理后再合并结果 -
位运算优化：
如果数值范围有限，可以利用位运算特性 -
SIMD指令：
使用现代CPU的并行指令加速计算
在实际比赛中，理解题目本质并找到数学规律往往比盲目优化代码更有效。这道题教会我们，有时候退一步从数学角度思考问题，反而能找到更优的解决方案。