类似于二分查找,三分搜索法也是比较常用的基于分治思想的高效查找方法。但是和二分不同,二分只适用于单调函数,比如常用的对单调递增或单调递减的一个序列中的某一个元素进行查找,三分却突破了这种限制,可以用于左边递增右边递减或者相反的,这么一类函数,也就是常说的凸函数和凹函数。但是为什么三分法可以用于凸函数或者凹函数呐,这其实是因为这种函数总是有一个最大值或者最小值,这样就可以借此判断出三分法中两个中点相对相对于极值的位置。
三分搜索的实现主要是判断midl和midr所在值的大小。以凸函数为例(凹函数类似,只是判mid大小的时候保留小的即可(其实也是保留离极值最近的mid)),先以left和right为端点计算出它们的中点midl,然后再以midl和right为端点计算出它们的中点midr,接下来就需要判断f(midl)和f(midr)值的大小了,如果f(midl)大于f(midr),那么说明midl靠近极值,此时令right=midr,否则说明midr靠近极值,此时则令left=midl,总之就是要保留离极值最近的那一个mid,然后重复前面的过程,直到left和right十分接近,最终f(left)就等于了极值。
https://www.cometoj.com/contest/33/problem/B?problem_id=1456
#include <iostream>
#include<stdio.h>
#include<bits/stdc++.h>
#include<algorithm>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
ll n,m,s[200005];
ll solve(ll mid)
{
ll ans=0;
for(ll i=0; i<n; i++)
{
if(s[i]>=mid)
ans+=(s[i]-mid)*(s[i]-mid);
else
ans+=max(mid-s[i]-m,(ll)0)*max(mid-s[i]-m,(ll)0);
}
return ans;
}
int main()
{
cin>>n>>m;
for(ll i=0; i<n; i++)
cin>>s[i];
ll l=1,r=2e5;
while(r-l>1)
{
ll midl=(l+r)/2;
ll midr=(midl+r)/2;
if(solve(midl)>=solve(midr))
l=midl;
else
r=midr;
}
ll ans=solve(l);
ans=min(ans,solve(r));
cout<<ans<<endl;
return 0;
}