牛客寒假算法基础集训营1-H.牛牛看云

标签：枚举，二分，前缀和

思路

本题本质上就是计算一个式子的值，所以算法有很多，这里提供两种思路。

第一种思路是枚举，当然不是指直接两层for循环O(n^2)的算法，因为这样会超时，注意到ai的范围很小，而且由于要计算式子的特殊性，使得我们不需要枚举每个数，而只需要知道这个数出现的次数同样可以计算出答案，那枚举范围就变成了[0,1000]，这样复杂度是完全够的，需要注意的是计算的两个数相等和不相等时的次数计算是不一样的，具体解释在代码中

第二种思路是二分+前缀和，式子中的ai和aj的地位是一样的（即|ai+aj-1000|=|(ai-500)+(aj-500)|），那么我们可以在输入后直接每个数都减去500，方便后续计算，然后将处理后的数列从小到大排序（顺序改变不会改变结果）并求前缀和，这样操作能将序列分成正负两部分，然后只需要枚举每一个数，然后二分找到它的相反数所在的位置（即和为0的位置），该位置左边的数与ai的和肯定小于零，所以要加一个负号，右边的数与ai的和大于0，直接加上就可以了，别忘了还有自身加自身的绝对值

代码实现

思路理清之后，代码实现就比较简单了，注意答案开long long即可
第一种思路：

#include<stdio.h>
#include<iostream>
#include<math.h>
using namespace std;
int n;
long long cnt[1005];
int main(){
    cin>>n;
    int x;
    long long ans=0;
    for(int i=1;i<=n;i++){
    	cin>>x;
    	cnt[x]++;                //存每个数出现的次数
	}
	long long t=0;
	for(int i=0;i<=1000;i++){
		for(int j=i;j<=1000;j++){
			if(i==j) //相等时除了加上自身外，还要加上的相当于cnt个数中
				t=cnt[i]+cnt[i]*(cnt[i]-1)/2; //选两个数的组合
			else{                                  
				t=cnt[i]*cnt[j];   //不相等时直接将次数相乘
			}                      //次数计算方法手动模拟一下就能理解了
			ans+=t*abs(i+j-1000);
		}
	}
	cout<<ans;
	return 0;
}

第二种思路：

#include<stdio.h>
#include<iostream>
#include<math.h>
#include<cmath>
#include<algorithm>
using namespace std;
int n;
int a[1000005];
int sum[1000005];
int main(){
    cin>>n;
    long long ans=0;
    for(int i=1;i<=n;i++){
    	cin>>a[i];
    	a[i]-=500;
	}
	sort(a+1,a+1+n);
	for(int i=1;i<=n;i++){
		sum[i]=sum[i-1]+a[i];   //前缀和处理
	}
	for(int i=1;i<=n;i++){
		int now=lower_bound(a+i,a+1+n,-a[i])-a;//这里是找相反数的位置，偷懒直接用现成的了
		ans=ans+(n-now)*a[i]+sum[n]-sum[now]+abs(2*a[i])-((now-i)*a[i]+sum[now]-sum[i]);
	}
	cout<<ans;
	return 0;
}