题意:有一个数组,长度为n,把它的每一位加上a或者是减去a(a>=0)
问它的所有元素平方和最小为多少,以分数表示
思路“很明显,要找到a,使得每一个元素尽量小,开始我是用了模拟去做,我觉得是没有什么大问题,样例过了,但是还是wa了。
后来看了题解,才发现要推导公式,用二次函数的知识去做
对于每一个元素wi:
如果wi是负的,就应该加上a。(wi+a)^2=wi^2 + 2*wi*a + a^2,因为wi是负的,所以+2*wi*a可以写成-(|wi|*a)
如果wi是正的,就应该减去a。(wi-a)^2=wi^2- 2*wi *a + a^2。因为wi是正的,也可以写成-(|wi|*a)
用sum代表所有wi的绝对值的和
所有的元素平方加起来,∑(wi^2)+ n*a^2 - 2*sum*a
∑(wi^2)是一个常数,当上式是关于a的一元二次方程,当a=sum/n的时候有最小值
最小值:-sum^2/n + ∑(wi^2)
把n当作分母,求出-sum^2+∑(wi^2)*n就行了
代码:
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<set>
#include<queue>
#include<map>
using namespace std;
const int MAXN=10000007;
long long a[100005],b[100005];
long long gcd(long long x,long long y)
{
if(y==0)return x;
else return gcd(y,x%y);
}
int main()
{
long long T,i,s,n,g,sum,t,len;
scanf("%lld",&T);
while(T--)
{
scanf("%lld",&n);
for(i=1;i<=n;i++)
scanf("%lld",&a[i]);
sum=0;t=0;
for(i=1;i<=n;i++)
{
t+=(a[i]*a[i]);
if(a[i]>0)
sum+=a[i];
else sum-=a[i];//sum为数组元素绝对值之和
}
t*=n;//t为元素平方和
sum*=sum;
t-=sum;
if(t==0)
{
printf("0/1\n");
}
else
{
if(t>n)g=gcd(t,n);
else g=gcd(n,t);
//cout<<g<<endl;
t/=g;
n/=g;//约分
printf("%lld/%lld\n",t,n);
}
}
return 0;
}