CSP-J复赛冲刺必刷题 | P2671 求和

学习C++从娃娃抓起！记录下CSP-J备考学习过程中的题目，记录每一个瞬间。

附上汇总贴：CSP-J复赛冲刺必刷题 | 汇总_热爱编程的通信人的博客-CSDN博客

---

【题目描述】

一条狭长的纸带被均匀划分出了n个格子，格子编号从1到n。每个格子上都染了一种颜色colori用[1,m]当中的一个整数表示），并且写了一个数字numberi。

定义一种特殊的三元组：(x,y,z)，其中x,y,z都代表纸带上格子的编号，这里的三元组要求满足以下两个条件：

1. xyz是整数,x<y<z,y−x=z−y
2. colorx=colorz

满足上述条件的三元组的分数规定为(x+z)×(numberx+numberz)。整个纸带的分数规定为所有满足条件的三元组的分数的和。这个分数可能会很大，你只要输出整个纸带的分数除以10,007所得的余数即可。

【输入】

第一行是用一个空格隔开的两个正整数n和m,n表纸带上格子的个数，m表纸带上颜色的种类数。

第二行有n用空格隔开的正整数，第i数字number表纸带上编号为i格子上面写的数字。

第三行有n用空格隔开的正整数，第i数字color表纸带上编号为i格子染的颜色。

【输出】

一个整数，表示所求的纸带分数除以10007所得的余数。

【输入样例】

6 2
5 5 3 2 2 2
2 2 1 1 2 1

【输出样例】

82

【代码详解】

/*
i*((cnt-2)*ai + sum)
*/
#include <bits/stdc++.h>
using namespace std;
long long cnt[100005][2], sum[100005][2];
long long a[100005];
long long col[100005];
int main()
{
    int n, m;
    cin >> n >> m;
    for (int i=1; i<=n; i++) {
        cin >> a[i];
    }
    for (int i=1; i<=n; i++) {
        cin >> col[i];
        cnt[col[i]][i%2]++;  //统计这个颜色奇偶位置的个数
        sum[col[i]][i%2]+=a[i];  //统计这个颜色在奇偶位置的和
    }
    long long ans = 0;
    for (int i=1; i<=n; i++) {
        int c = col[i];
        ans += (i*((cnt[c][i%2]-2+10007)%10007*a[i]%10007 + sum[c][i%2]%10007))%10007;
        ans %=10007;
    }
    cout << ans;
    return 0;
}

【运行结果】

6 2
5 5 3 2 2 2
2 2 1 1 2 1
82