一开始想用贪心做,找出最高与最低的山峰求出它俩的差再将这个差除以2(考虑是奇数还是偶数的情况)最低峰加上一个值最高峰减去一个值,再排序,重复上述操作直到最高与最低相差不大于17。不知道为什么这样做不对。
看的题解是用枚举做的,因为高度是大于等于一小于等于一百,就从1到83遍历,那么就找出输入的高度中不满足 >=i && <=i+17 的,求出最小值。
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <string>
#include <cstring>
#include <vector>
#include <map>
#include <set>
#include <queue>
#define ll long long
#define lowbit(x) ((~x+1)&x)
using namespace std;
const int N = 2e5+10;
const ll inf = 0x3f3f3f3f;
int n,m,k;
int main() {
int n,h[10005];
cin>>n;
for(int i=1; i<=n; i++) {
cin>>h[i];
}
ll ans=1<<30;
for(int k=1; k<=83; k++) {
int l=k,r=k+17;
ll cmp=0;
for(int i=1; i<=n; i++) {
if(h[i]<l) cmp+=(l-h[i])*(l-h[i]);
else if(h[i]>r) cmp+=(h[i]-r)*(h[i]-r);
}
ans = min(ans,cmp);
}
cout<<ans<<endl;
return 0;
}