Codeforces Round #635 (Div. 2) D. Xenia and Colorful Gems（二分）

题目链接


题意：要你从数组r中选一个数x，从数组g中选一个数y，从数组b中选一个数z，使得（x-y）^2+(x-z) ^2+(y-z) ^2最小。
思路：一开始想复杂了，想着要如何确定x的情况下使结果最小，其实要最小的话肯定是x<y<z，同时x和z离y很近（注意一下这只是举例并不是最终结论），于是我们发现所有的情况都可以写成x<y<z的情况，在确定y的情况用二分查找里y最近的x和z，但是最终答案并一定是x<y<z，有可能是y<x<z，y<z<x等等都有可能，所以我们要枚举所有的情况才不会疏漏。

#include<bits/stdc++.h> 
using namespace std;
typedef long long ll;
const int maxn=1e5+1; 
ll a[maxn],b[maxn],c[maxn];
ll get(ll x,ll y,ll z)
{
	return (x-y)*(x-y)+(x-z)*(x-z)+(y-z)*(y-z);
}
ll check(ll a[],int na,ll b[],ll nb,ll c[],int nc)
{
	ll x,y,z,ans=2e18;
	for(int i=1;i<=na;++i)
	{
		 x=a[i];
		int pos=lower_bound(b+1,b+1+nb,x)-b;
		if(b[nb]<x||b[pos]<x) continue;
		 y=b[pos];
		int l=1,r=nc;
		if(c[1]>x) continue;
		while(l<=r)
		{
			int mid=(l+r)>>1;
			if(c[mid]<=x) l=mid+1;
			else r=mid-1;
		}
		if(c[r]<=x) z=c[r];
		ans=min(ans,get(x,y,z));
	}
	return ans;
}
int main()
{
	int T,na,nb,nc;
	scanf("%d",&T);
	while(T--)
	{
		ll t,ans=2e18;
		scanf("%d %d %d",&na,&nb,&nc);
		for(int i=1;i<=na;++i) scanf("%lld",&a[i]);
		for(int i=1;i<=nb;++i) scanf("%lld",&b[i]);
		for(int i=1;i<=nc;++i) scanf("%lld",&c[i]);
		sort(a+1,a+1+na);
		sort(b+1,b+1+nb);
		sort(c+1,c+1+nc);
		ans=min(ans,check(a,na,b,nb,c,nc));ans=min(ans,check(a,na,c,nc,b,nb));
		ans=min(ans,check(b,nb,a,na,c,nc));ans=min(ans,check(b,nb,c,nc,a,na));
		ans=min(ans,check(c,nc,a,na,b,nb));ans=min(ans,check(c,nc,b,nb,a,na));
		printf("%lld\n",ans);
	}
}