题目描述 https://www.luogu.org/problemnew/show/P3650
暴力,枚举高度范围,将需要修改的山按照题目中的算法相加,取最小值。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int a[1005];
long long minn=999999999999,ans;
int main()
{ int n;
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
sort(a+1,a+n+1);
int d=a[n]-a[1];
if(d<=17){ cout<<0; return 0;}
d=d-17;
for(int i=1;i<=d;i++)
{
int l=a[1]+i,r=a[n]-d+i;ans=0;
for(int j=1;j<=n;j++)
{
if(a[j]<l) ans=ans+(l-a[j])*(l-a[j]);
if(a[j]>=l) break;
}
for(int j=n;j>=1;j--)
{
if(a[j]>r) ans=ans+(a[j]-r)*(a[j]-r);
if(a[j]<=l) break;
}
minn=min(ans,minn);
}
printf("%lld",minn);
return 0;
}