P2671 [NOIP2015 普及组] 求和

题目描述

一条狭长的纸带被均匀划分出了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;
}