53 lines
1.7 KiB
C++
53 lines
1.7 KiB
C++
|
#include <bits/stdc++.h>
|
|||
|
|
using namespace std;
|
|||
|
|
|
|||
|
|
const int N = 100010, mod = 10007;
|
|||
|
|
|
|||
|
|
int w[N]; // 第i个格子的数字
|
|||
|
|
int c[N]; // 第i个格子的颜色
|
|||
|
|
int s[N][2];
|
|||
|
|
// s[i][0]:颜色为i、编号为偶数格子上数字的和
|
|||
|
|
// s[i][1]:颜色为i、编号为奇数格子上数字的和
|
|||
|
|
|
|||
|
|
int cnt[N][2];
|
|||
|
|
// cnt[i][0]:颜色为i、编号为偶数格子的个数
|
|||
|
|
// cnt[i][1]:颜色为i、编号为奇数格子的个数
|
|||
|
|
|
|||
|
|
int main() {
|
|||
|
|
int n, m;
|
|||
|
|
cin >> n >> m;
|
|||
|
|
for (int i = 1; i <= n; i++) cin >> w[i]; // 第i个格子的数字
|
|||
|
|
// 预处理
|
|||
|
|
for (int i = 1; i <= n; i++) { // 遍历每个格子
|
|||
|
|
cin >> c[i]; // 格子颜色c[i]
|
|||
|
|
/*
|
|||
|
|
装入不同的组中,组划分是两个规则：
|
|||
|
|
① 颜色必须相同
|
|||
|
|
② 奇偶性必须相同
|
|||
|
|
所以，c[i]相同的放到同一个颜色组内，并且，在同一个颜色组内，奇偶数还必须相同。
|
|||
|
|
|
|||
|
|
s[][]:随着i的不断向后遍历，s中记录了相同颜色，相同奇偶性的格子，数字的累加和
|
|||
|
|
cnt[][]:记录每个分组中的格子个数
|
|||
|
|
*/
|
|||
|
|
s[c[i]][i % 2] = (s[c[i]][i % 2] + w[i]) % mod; // 累加分组内数字和
|
|||
|
|
cnt[c[i]][i % 2]++; // 维护分组内格子个数
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int ans = 0;
|
|||
|
|
for (int i = 1; i <= n; i++) // 枚举每个格子
|
|||
|
|
/*
|
|||
|
|
Q:这个格子在哪个分组里呢？
|
|||
|
|
答：
|
|||
|
|
(1) c[i] : 按颜色划分
|
|||
|
|
(2) i % 2 : 按奇偶性划分
|
|||
|
|
|
|||
|
|
Q:这个分组中格子的数量是多少呢？
|
|||
|
|
答： cnt[c[i]][i % 2]
|
|||
|
|
|
|||
|
|
*/
|
|||
|
|
ans = (ans + i * ((cnt[c[i]][i % 2] - 2) * w[i] % mod + s[c[i]][i % 2])) % mod;
|
|||
|
|
|
|||
|
|
printf("%d\n", ans);
|
|||
|
|
return 0;
|
|||
|
|
}
|