1235. 求和
题目描述
一条狭长的纸带被均匀划分出了 \(n\) 个格子,格子编号从 \(1\) 到 \(n\)。每个格子上都染了一种颜色 \(color_i\) 用 \([1,m]\) 当中的一个整数表示),并且写了一个数字 \(number_i\)。
编号 | 1 | 2 | 3 | 4 | 5 | 6 |
---|---|---|---|---|---|---|
颜色和数字 | \(\color{blue}{5}\) | \(\color{blue}{5}\) | \(\color{red}{3}\) | \(\color{red}{2}\) | \(\color{blue}{2}\) | \(\color{red}{2}\) |
定义一种特殊的三元组:\((x,y,z)\),其中 \(x,y,z\) 都代表纸带上格子的编号,这里的三元组要求满足以下两个条件:
\(x,y,z\) 都是整数,\(x<y<z,y-x=z-y\)。
\(color_x=color_z\)。
满足上述条件的三元组的分数规定为 \((x+z) \times (number_x+number_z)\)。整个纸带的分数规定为所有满足条件的三元组的分数的和。这个分数可能会很大,你只要输出整个纸带的分数除以 \(10007\) 所得的余数即可。
输入
第一行,是用一个空格隔开的两个正整数 \(n\) 和 \(m\),\(n\) 代表纸带上格子的个数,\(m\) 代表纸带上颜色的种类数。
第二行,有 \(n\) 个用空格隔开的正整数,第 \(i\) 个数字代表纸带上编号为 \(i\) 的格子上面写的数字 \(number_i\)。
第三行,有 \(n\) 个用空格隔开的正整数,第 \(i\) 个数字代表纸带上编号为 \(i\) 的格子染的颜色 \(color_i\)。
输出
一个整数,表示所求的纸带分数除以 \(10,007\) 所得的余数。
样例1
输入
6 2
5 5 3 2 2 2
2 2 1 1 2 1
输出
82
解释
纸带如题目描述中的图所示。
所有满足条件的三元组为:\((1, 3, 5), (4, 5, 6)\)。
所以纸带的分数为 \((1 + 5) \times (5 + 2) + (4 + 6) \times (2 + 2) = 42 + 40 = 82\)。
样例2
输入
15 4
5 10 8 2 2 2 9 9 7 7 5 6 4 2 4
2 2 3 3 4 3 3 2 4 4 4 4 1 1 1
输出
1388
数据范围限制
对于第 \(1\) 组至第 \(2\) 组数据,\(1 \leq n \leq 100\),\(1 \leq m \leq 5\);
对于第 \(3\) 组至第 \(4\) 组数据,\(1 \leq n \leq 3000\),\(1 \leq m \leq 100\);
对于第 \(5\) 组至第 \(6\) 组数据,\(1 \leq n \leq 10^5\),\(1 \leq m \leq 10^5\),且不存在出现次数超过 \(20\) 的颜色;
对于全部 \(10\) 组数据,\(1 \leq n \leq 10^5\), \(1 \leq m \leq 10^5\), \(1 \leq color_{i} \leq m\), \(1 \leq number_{i} \leq 10^5\)。
来源
NOIP2015 普及组 T3
信息
- ID
- 1234
- 难度
- (无)
- 分类
- (无)
- 标签
- (无)
- 递交数
- 0
- 已通过
- 0
- 通过率
- ?
- 上传者