模拟赛20210810A T1

题目

大意：
给定n个数$a_1$~$a_n$
求任意一对数（$a_i,a_j$） 的 $|(i-j)\ast (ai-aj)|$之和。

思路

显然只有当$a_i,a_j$不同时，才会对答案有贡献，所以只用让每个1和每个0配对。
当很多个1（$i_1,i_2\cdot \cdot \cdot \cdot \cdot \cdot i_m$）和一个0（$j$）相配对时，
产生$(i_1-j)+(i_2-j)\cdot \cdot \cdot \cdot \cdot \cdot +(i_m-j)$
即产生$(i_1+i_2\cdot \cdot \cdot \cdot \cdot \cdot +i_m)-(m\ast j)$ 的贡献。
不过每次$i-j$的计算是加绝对值的，也就是分1在0的左边 和 1在0的右边两种情况。
所以要分类讨论：
用前缀和预处理$(i_1+i_2\cdot \cdot \cdot \cdot \cdot \cdot +i_m)$和 $m$，再将数组反着做一遍前缀和（不是等于后缀和）
其中$m$是一个数组，$m_k$表示前$k$位（或后$k$位）有多少个1。
这样，正着的前缀和可以O(1)求出对于一个0，它的右边的所有1的贡献；反着的O(1)求出左边的所有1的贡献。
枚举每个0，将正反（左右）的结果相加，即可$O(n)$得出最终答案

代码

#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
const int M=1001000;
int n,a[M];
long long s1[M],s2[M],b[M],c[M],ans;
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;++i)
	{
		scanf("%d",&a[i]);
		s1[i]=s1[i-1]+(i*a[i]); //位置前缀和预处理
		b[i]=b[i-1]+a[i]; // 1的个数前缀和预处理
	}
	//反着的预处理，类上
	for(int i=n;i>=1;--i)
	{
		s2[i]=s2[i+1]+((n-i+1)*a[i]);
		c[i]=c[i+1]+a[i];
	}
	for(int i=1;i<=n;++i)
	{
		if(a[i]==0)
		{
			ans+=s1[n]-s1[i]-((b[n]-b[i])*i); //右边的所有1的贡献
			ans+=s2[1]-s2[i]-((c[1]-c[i])*(n-i+1)); //右边的所有1的贡献
		}
	}
	cout<<ans;
}