AtCoder Beginner Contest 194 C - Squared Error

最新推荐文章于 2025-03-18 12:59:01 发布

木落淮南，雨晴雲夢

最新推荐文章于 2025-03-18 12:59:01 发布

阅读量303

点赞数

分类专栏： AT

本文链接：https://blog.csdn.net/lcl1234567890lcl/article/details/114476232

版权

区间值计数排列组合非主流解法 C++实现

关键词由CSDN通过智能技术生成

AT 专栏收录该内容

21 篇文章

订阅专栏

传送门
在这里插入图片描述

看到别人的都是前缀和，我就来发一个不太主流的题解吧

观察发现，A的取值在[-200,200]，这个区间不算大，因此就可以记录每一个值的个数，再通过排列组合的方式求解,通过i和j的值是否相等进行判断，然后求解

#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#include <bits/stdc++.h>
#define inf 0x7fffffff
#define ll long long
#define int long long
//#define double long double
#define eps 1e-8
//#define mod 1e9+7
using namespace std;
const int mod=1e9+7;
const int M=2e3+5;
const int N=2*1e6+5;//?????????? 4e8
int n;
map < int , int > mp;
int a[N];
int ans;
int c(int x)
{
	return x*(x-1)/2;
}
signed main()
{
//	ios::sync_with_stdio(false);
	cin>>n;
	for(int i=1;i<=n;i++)  scanf("%lld",&a[i]),mp[a[i]]++;
	for(int i=-200;i<=200;i++)  for(int j=i;j<=200;j++)
		if(i==j)  ans+=(n-1)*i*i*mp[i]-i*i*c(mp[i])*2;
		else  ans-=2*i*j*mp[i]*mp[j];
	cout<<ans<<endl;
	return 0;
}