Codeforces Xenia and Colorful Gems(二分查找、枚舉）

題目大意：
給定三個數組，從三個數組中分別選出一個數字，使得要求的結果表達式最小

解題思路：
由題意可知三個數x<=y<=z，枚舉y，二分查找找到最接近與y的x與z，求結果，取所有結果的最小值

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1e5+10;
ll ans;
int n,t;
void solve(vector<int> a,vector<int> b,vector<int> c)
{
    for(auto y:b)
    {
        auto x=upper_bound(c.begin(),c.end(),y);
        auto z=lower_bound(a.begin(),a.end(),y);
        if(x==c.begin()||z==a.end()) continue;
        x--;
        ans=min(ans,(ll)(y-*x)*(y-*x)+(ll)(y-*z)*(y-*z)+(ll)(*x-*z)*(*x-*z));
    }
}
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        int n1,n2,n3;
        scanf("%d%d%d",&n1,&n2,&n3);
        vector<int> a(n1),b(n2),c(n3);
        for(int i=0;i<n1;++i) scanf("%d",&a[i]);
        for(int i=0;i<n2;++i) scanf("%d",&b[i]);
        for(int i=0;i<n3;++i) scanf("%d",&c[i]);
        ans=0x3f3f3f3f3f3f3f3f;
        sort(a.begin(),a.end());
        sort(b.begin(),b.end());
        sort(c.begin(),c.end());
        solve(a,b,c);
        solve(a,c,b);
        solve(b,a,c);
        solve(b,c,a);
        solve(c,a,b);
        solve(c,b,a);
        printf("%lld\n",ans);
    }
    return 0;
}