P2671 [NOIP2015 普及组] 求和

[NOIP2015 普及组] 求和

题目

题目背景

NOIP2015 普及组 T3

题目描述

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

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

1. $xyz$是整数,$x<y<z,y-x=z-y$

2. $colorx=colorz$

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

输入格式

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

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

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

输出格式

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

样例 #1

样例输入 #1

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

样例输出 #1

82

样例 #2

样例输入 #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+5)\times (5+2)+(4+6)\times (2+2)=42+40=82$。

对于第 $1$ 组至第 $2$ 组数据， $1\le n\le 100,1\le m\le 5$；

对于第$ 3$ 组至第 $4$ 组数据， $1\le n\le 3000,1\le m\le 100$；

对于第 $5$ 组至第$ 6 $组数据， $1\le n\le 100000,1\le m\le 100000$，且不存在出现次数超过$ 20 $的颜色；

对 于 全 部 $10$ 组 数 据 ， $1\le n\le 100000,1\le m\le 100000,1\le colo{r}_i\le m,1\le numbe{r}_i\le 100000$

---

题解

注意观察 x, y, z 之间的关系，由于最终答案的计算与 y 无关，而 x < y < z, y-x = z-y，因此 y 的作用仅仅是保证 x 和 z 奇偶性相同。
再来看一下 x 和 z 的答案，$(x+z)\times (numbe{r}_x+numbe{r}_z)$ 可以拆分成 $x{n}_x+z{n}_z+x{n}_z+z{n}_x$，这样方便我们统计其对答案的贡献。
我们将奇偶格子分开，对于相同的颜色，我们可以枚举每个 $z$ ，累计所有 $z$ 之前颜色相同的贡献四种贡献，再累加起来就可以了。
我们不妨设置四个桶分别存储这四个贡献：c1存储所有满足相同颜色的 $x{n}_x$ 相加之和，c2存储每个 $x$ 之前与其颜色相同的格子个数，c3存储每个 $x$ 之前与其颜色相同的编号的和，c4存储每个 $x$ 之前与其颜色相同的 $n$ 的和。
最后分别统计奇偶两种格子，统计出答案即可。
Code：

#include <iostream>
#include <algorithm> 
#include <cstring> 
using namespace std;
typedef long long ll;
const int maxn = 1e5 + 5;
const int mod = 10007;
int n, m, ans;
ll num1[maxn], num2[maxn], col1[maxn], col2[maxn];//num1，col1存储奇数，num2，col2存储偶数
ll cnt1[maxn], cnt2[maxn], cnt3[maxn], cnt4[maxn];//每个桶都按照不同颜色分开
ll odd, enn;//奇数和偶数的个数
int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	
	cin >> n >> m;
	for(int i = 1; i <= n; i++) {
		if(i&1) cin >> num1[++odd];
		else cin >> num2[++enn];
	}
	odd = enn = 0;
	for(int i = 1; i <= n; i++) {
		if(i&1) cin >> col1[++odd];
		else cin >> col2[++enn];
	}
	for(int i = 1; i <= odd; i++) {//先计算奇数
		ll pos = i*2-1, c = col1[i];//因为i不是原来数字下标，所以要重新计算。
		ll x = pos*num1[i] % mod;//x即下标和值的积
		ans = (ans + cnt1[c] + cnt2[c]*x%mod + cnt3[c]*num1[i]%mod + cnt4[c]*pos%mod) % mod;
		cnt1[c] = (cnt1[c] + x) % mod;//xn_x
		cnt2[c]++;//zn_z中的z
		cnt3[c] += pos;//xn_z中的x
		cnt4[c] += num1[i];//zn_x中的n_x
	}
	memset(cnt1, 0, sizeof(cnt1));//清空统计偶数
	memset(cnt2, 0, sizeof(cnt2));
	memset(cnt3, 0, sizeof(cnt3));
	memset(cnt4, 0, sizeof(cnt4));
	for(int i = 1; i <= enn; i++) {
		ll pos = i*2, c = col2[i];
		ll x = pos*num2[i] % mod;
		ans = (ans + cnt1[c] + cnt2[c]*x%mod + cnt3[c]*num2[i]%mod + cnt4[c]*pos%mod) % mod;
		cnt1[c] = (cnt1[c] + x) % mod;
		cnt2[c]++;
		cnt3[c] += pos;
		cnt4[c] += num2[i];
	}
	cout << ans;//输出结果
    return 0;
}