一条狭长的纸带被均匀划分出了 n� 个格子,格子编号从 11 到 n�。每个格子上都染了一种颜色 colori������(颜色用 [1,m][1,�] 当中的一个整数表示),并且写了一个数字 numberi�������。
定义一种特殊的三元组:(x,y,z)(�,�,�),其中 x,y,z�,�,� 都代表纸带上格子的编号,这里的三元组要求满足以下两个条件:
-
x,y,z�,�,� 是整数,x<y<z,y−x=z−y�<�<�,�−�=�−�;
-
colorx=colorz������=������。
满足上述条件的三元组的分数规定为 (x+z)×(numberx+numberz)(�+�)×(�������+�������)。整个纸带的分数规定为所有满足条件的三元组的分数的和。这个分数可能会很大,你只要输出整个纸带的分数除以 1000710007 所得的余数即可。
输入格式
第一行是用一个空格隔开的两个正整数 n� 和 m�,n� 代表纸带上格子的个数,m� 代表纸带上颜色的种类数。
第二行有 n� 个用空格隔开的正整数,第 i� 个数字 numberi������� 代表纸带上编号为 i� 的格子上面写的数字。
第三行有 n� 个用空格隔开的正整数,第 i� 个数字 colori������ 代表纸带上编号为 i� 的格子染的颜色。
输出格式
一行一个整数,表示所求的纸带分数除以 1000710007 所得的余数。
样例输入1
6 2
5 5 3 2 2 2
2 2 1 1 2 1
样例输出1
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
样例输出2
1388
样例 1 说明
纸带如题目描述中的图所示。
所有满足条件的三元组为: (1,3,5),(4,5,6)(1,3,5),(4,5,6)。
所以纸带的分数为 (1+5)×(5+2)+(4+6)×(2+2)=42+40=82(1+5)×(5+2)+(4+6)×(2+2)=42+40=82。
数据范围
对于 100%100% 的数据,保证 1≤n≤100000,1≤m≤100000,1≤colori≤m,1≤numberi≤1000001≤�≤100000,1≤�≤100000,1≤������≤�,1≤�������≤100000。
#include <bits/stdc++.h>
using namespace std;
const int N = 100005, P = 10007;
int a[N], c[N];
int sum[4][2][N];
int l[2][N];
int main() {
int n, m, res = 0;
cin >> n >> m;
for (int i = 1; i <= n; ++i) {
cin >> a[i];
}
for (int i = 1; i <= n; ++i) {
cin >> c[i];
}
for (int i = 1; i <= n; ++i) {
int A = sum[0][i % 2][l[i % 2][c[i]]] % P;
int B = sum[1][i % 2][l[i % 2][c[i]]] % P;
int C = sum[2][i % 2][l[i % 2][c[i]]] % P;
int D = sum[3][i % 2][l[i % 2][c[i]]] % P;
if (l[i % 2][c[i]]) {
res += A % P * a[i] % P +B % P * i % P +C % P +D % P * i % P * a[i] % P;
}
res = res % P;
sum[0][i % 2][i] = A % P + i % P;
sum[1][i % 2][i] = B % P + a[i] % P;
sum[2][i % 2][i] = C % P + i % P * a[i] % P;
sum[3][i % 2][i] = D + 1;
l[i % 2][c[i]] = i;
}
cout << res % P << '\n';
return 0;
}