题目描述
一条狭长的纸带被均匀划分出了nnn个格子,格子编号从111到nnn。每个格子上都染了一种颜色color_icolor_icolor_i用[1,m][1,m][1,m]当中的一个整数表示),并且写了一个数字number_inumber_inumber_i。
定义一种特殊的三元组:(x,y,z)(x,y,z)(x,y,z),其中x,y,zx,y,zx,y,z都代表纸带上格子的编号,这里的三元组要求满足以下两个条件:
xyz是整数,x<y<z,y−x=z−y
color-x=color-z
满足上述条件的三元组的分数规定为(x+z)×(number_x+number_z)(x+z) \times (number_x+number_z)(x+z)×(number_x+number_z)。整个纸带的分数规定为所有满足条件的三元组的分数的和。这个分数可能会很大,你只要输出整个纸带的分数除以10,00710,00710,007所得的余数即可。
输入格式
第一行是用一个空格隔开的两个正整数nnn和m,nm,nm,n表纸带上格子的个数,mmm表纸带上颜色的种类数。
第二行有nnn用空格隔开的正整数,第iii数字numbernumbernumber表纸带上编号为iii格子上面写的数字。
第三行有nnn用空格隔开的正整数,第iii数字colorcolorcolor表纸带上编号为iii格子染的颜色。
输出格式
一个整数,表示所求的纸带分数除以100071000710007所得的余数。
输入输出样例
输入 #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
#include <bits/stdc++.h>
#define ll long long
const ll int MAX_N = 2e5 + 5;
#define bug(x) cout << #x << " == " << x << '\n';
using namespace std;
ll mapp[MAX_N][3] = {0};
ll a[MAX_N], c[MAX_N];
map<ll, ll> sl;
int main()
{
ll n, x;
cin >> n >> x;
for (register int i = 1; i <= n; i++)
{
scanf("%lld", &a[i]);
}
for (register int i = 1; i <= n; i++)
{
scanf("%lld", &c[i]);
mapp[c[i]][i % 2] += a[i];
mapp[c[i]][i % 2] %=10007;
sl[c[i] * 10000000 + (i % 2)]++;//记录颜色为c[i],
//sl[c[i] + (i % 2) * 1000000]%=10007;
}
ll sum = 0;
for (int i = 1; i <= n; i++)
{
sum += i * ((sl[c[i] + (i % 2) * 1000000] - 2) * a[i] % 10007 + mapp[c[i]][i % 2]) % 10007;
sum%=10007;
// bug(sum);
}
printf("%lld\n",sum%10007);
//system("pause");
return 0;
}