2671 求和

2671 求和

首先需要找出满足条件的三元组，最后将三元组通过求和算出来，一般这种题，都少不了前缀和
不过这个题，好像是一个方程推导题，这种题，我练的不好
不过

任何一个算法，都是对暴力的优化
by 警察浩翔

所以，如果我们一个一个枚举x，y，z时间复杂度肯定会炸对吧
我们可以得到，2y=z+x
由此可知z+x肯定是一个偶数，z和x必须同为偶数或者同为奇数
所以只需要枚举z和x就好了，不过还会超时
我们可以利用分组的思想，将每一个颜色进行分组
然后按照奇偶分组，所以一共有2m组
假设一个分组有k个数，这个分组中的数分别是x[1]，x[2]，x[3]…x[k]，下标是y[1]，y[2]，y[3]…y[k]
所以一步一步推导
x[1](y[1]+y[2]+y[3]+…y[k])+x[2] (y[1]+y[2]+y[3]+…y[k])
=x[1](y[1](n-2)+y[1]+y[2]+…)+x[2](y[2](n-2)+y[1]+y[2]+…)
前缀和就体现出来了
所以到时候求出来y的前缀和就好了，时间复杂度就是O(n)

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
using namespace std;
const int SIZE=1e6+5;
int n,m;
int f[SIZE][2];
int x[SIZE];int c[SIZE];
int sum[SIZE][2];
int ans;
const int mod=10007;
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		cin>>x[i];
	}
	for(int i=1;i<=n;i++)
	{
		cin>>c[i];
		f[c[i]][i%2]++;
		sum[c[i]][i%2]=(sum[c[i]][i%2]+x[i])%mod;
	}
	for(int i=1;i<=n;i++)
	{
		ans=(ans+i*((f[c[i]][i%2]-2)*x[i]%mod+sum[c[i]][i%2]))%mod;
	}
	cout<<ans<<endl;
	return 0;
}