该代码段主要处理两个整数数组num和clr，使用模运算处理数组元素，并将元素按特定条件存储在idx和sum矩阵中。之后，通过对idx和sum的遍历计算得出结果ans。整个过程涉及数组索引操作和向量操作。
#include <bits/stdc++.h>
using namespace std;
int n, m, ans, num[100005], clr[100005], sum[100005][2];
vector<int> idx[100005][2];
const int MOD = 10007;
int main()
{
scanf("%d %d", &n, &m);
for (int i = 1; i <= n; i++)
{
scanf("%d", num + i);
num[i] %= MOD;
}
for (int i = 1; i <= n; i++)
{
scanf("%d", clr + i);
clr[i] %= MOD;
idx[clr[i]][i & 1].push_back(i); // idx记录下标
sum[clr[i]][i & 1] = (sum[clr[i]][i & 1] + num[i]) % MOD; // sum记录num的和
}
for (int i = 1; i <= m; i++)
for (int j = 0; j <= 1; j++)
{
int k = idx[i][j].size(); // k为总数
for (auto x : idx[i][j])
ans = (ans + (x * num[x] % MOD) * (k - 2) + x * sum[i][j]) % MOD; // 注意要一边求一边取模
}
printf("%d", ans);
return 0;
}
414
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
