Daimayuan Online Judge [NOIP2015 普及组] 求和

一条狭长的纸带被均匀划分出了 n� 个格子，格子编号从 11 到 n�。每个格子上都染了一种颜色 colori������（颜色用 [1,m][1,�] 当中的一个整数表示），并且写了一个数字 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)(�+�)×(�������+�������)。整个纸带的分数规定为所有满足条件的三元组的分数的和。这个分数可能会很大，你只要输出整个纸带的分数除以 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;
}