L i n k Link Link
D e s c r i p t i o n Description Description
一条狭长的纸带被均匀划分出了 n 个格子,格子编号从 1 到 n。每个格子上都染了一种颜色colori(用[1,m]当中的一个整数表示),并且写了一个数字numberi。
定义一种特殊的三元组:(x,y,z),其中x,y,z都代表纸带上格子的编号,这里的三元组要求满足以下两个条件:
1.x,y,z都是整数, x<y<z,y−x=z−y
2.colorx=colorz
满足上述条件的三元组的分数规定为(x + z) ∗ (numberx + numberz)。整个纸带的分数规定为所有满足条件的三元组的分数的和。这个分数可能会很大,你只要输出整个纸带的分数除以 10,007 所得的余数即可
I n p u t Input Input
第一行是用一个空格隔开的两个正整数 n 和 m,n 代表纸带上格子的个数,m 代表纸带上 颜色的种类数。
第二行有 n 个用空格隔开的正整数,第 i 个数字numberi代表纸带上编号为 i 的格子上面写的数字。
第三行有 n 个用空格隔开的正整数,第 i 个数字colori代表纸带上编号为 i 的格子染的颜色。
O u t p u t Output Output
共一行,一个整数,表示所求的纸带分数除以 10,007 所得的余数。
S a m p l e Sample Sample I n p u t Input Input
Sample Input1:
6 2
5 5 3 2 2 2
2 2 1 1 2 1
Sample Input2:
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
S a m p l e Sample Sample O u t p u t Output Output
Sample Output1:
82
Sample Output2:
1388
H i n t Hint Hint
【输入输出样例 1 说明】
纸带如题目描述中的图所示。
所有满足条件的三元组为:(1, 3, 5), (4, 5, 6)。
所以纸带的分数为(1 + 5) ∗ (5 + 2) + (4 + 6) ∗ (2 + 2) = 42 + 40 = 82。
对于第 1 组至第 2 组数据,1 ≤ n ≤ 100, 1 ≤ m ≤ 5;
对于第 3 组至第 4 组数据,1 ≤ n ≤ 3000, 1 ≤ m ≤ 100;
对于第 5 组至第 6 组数据,1 ≤ n ≤ 100000, 1 ≤ m ≤ 100000,且不存在出现次数超过 20 的颜色;
对于全部 10 组数据, 1 ≤ n ≤ 100000, 1 ≤ m ≤ 100000, 1 ≤ colori ≤ m, 1 ≤ numberi ≤ 100000。
S o l u t i o n Solution Solution
很容易想到40分暴力,枚举
x
x
x与
z
z
z,但是这样
O
(
n
2
)
O(n ^ 2)
O(n2)是会炸的,所以我们来讨论正解
以下是正解
因为
y
−
x
y - x
y−x ==
z
−
y
z - y
z−y,所以
x
x
x和
z
z
z的奇偶性相同
那么我们就把颜色分类,再按奇偶性分类
再把总数计算公式拆分开得到
x
∗
n
u
m
x
+
x
∗
n
u
m
z
+
z
∗
n
u
m
z
+
z
∗
n
u
m
x
x*num_x +x*num_z+z*num_z +z*num_x
x∗numx+x∗numz+z∗numz+z∗numx
然后我们枚举,就会得到很多个像上面一样的公式的运算,化简一下就能得到
后面的
∑
s
u
m
[
i
]
∗
∑
a
[
i
]
\sum sum[i] * \sum a[i]
∑sum[i]∗∑a[i]可以读入时预处理
时间复杂度就大大降低
C o d e Code Code
#include<iostream>
#include<cstdio>
#define ll long long
using namespace std;
ll n, m, ans;
ll x[100005], c[100005], s[100005][2], sum[100005][2];
int main()
{
scanf("%lld%lld", &n, &m);
for (int i = 1; i <= n; ++i)
scanf("%lld", &x[i]);
for (int i = 1; i <= n; ++i)
{
scanf("%lld", &c[i]);
s[c[i]][i % 2]++;
sum[c[i]][i % 2] = (sum[c[i]][i % 2] + x[i]) % 10007;
}
for (int i = 1; i <= n; ++i)
ans = (ans + i * ((s[c[i]][i % 2] - 2) * x[i] % 10007 + sum[c[i]][i % 2])) % 10007;
printf("%lld\n", ans);
return 0;
}