Codeforces1337 D. Xenia and Colorful Gems

题目链接
题目大意:
有a,b,c三个数组,从三个数组中各选出一个数,使得这个三个数ai,bj,ck的 ( a i − b j ) 2 + ( a i − c k ) 2 + ( b j − c k ) 2 (ai-bj)^2+(ai-ck)^2+(bj-ck)^2 (aibj)2+(aick)2+(bjck)2的值最小,输出这个最小值。

解题思路:
枚举中间这个数bj,则满足条件的数为ai ≤ \leq bj ≤ \leq ck。因为中间的数确定了,所以可以发现ai和ck越靠近bj则答案就会越小,所以ai就是小于等于bj的最大值,ck就是大于等于bj的最小值。因为不知道数组哪个作为中间值和前后值,所以需要枚举他们三个数组的排列组合。
实现方式可以使用二分O(nlogn),或者双指针O(n)。

解题代码:

#include<bits/stdc++.h>
using namespace std;
mt19937 rng_32(chrono::steady_clock::now().time_since_epoch().count());
typedef long long ll;
const int maxn=2e5+10;
int a[maxn],b[maxn],c[maxn];
ll ans=9e18;
ll getnum(ll a,ll b,ll c)
{
    ll t1=(a-b),t2=(a-c),t3=(b-c);
    return t1*t1+t2*t2+t3*t3;
}
void get1(int *a,int *b,int *c,int na,int nb,int nc)
{
	int i,j,k;
	for (j=0;j<nb;j++)
	{
		i=upper_bound(a,a+na,b[j])-a;
        //找到比bj大的数,然后下标-1 就是 ≤bj的数
        //注意不要越界了
		if (i-1<0 || (i==na && a[i-1]>b[j])) 
		continue ;
		i--;
		k=lower_bound(c,c+nc,b[j])-c;
		if (k==nc)
		continue;
		ans=min(ans,getnum(a[i],b[j],c[k]));
	}
}
void get(int *a,int *b,int *c,int na,int nb,int nc)
{
    int i=0,j=0,k=0;
    for(j=0;j<nb;j++)
    {
        while (i+1<na && a[i+1]<=b[j])
        i++;
        if(a[i]>b[j])
        continue;
        while (k<nc && c[k]<b[j])
        k++;
        if (k==nc)
        return ;
		ans=min(ans,getnum(a[i],b[j],c[k]));
    }
}
int main()
{
    int na,nb,nc;
    int cas;
    cin>>cas;
    while(cas--)
    {
        ans=9e18;
        scanf("%d%d%d",&na,&nb,&nc);
        for (int i=0;i<na;i++)
        scanf("%d",&a[i]);
        for(int i=0;i<nb;i++)
        scanf("%d",&b[i]);
        for (int i=0;i<nc;i++)
        scanf("%d",&c[i]);
        sort(a,a+na);
        sort(b,b+nb);
        sort(c,c+nc);
        get(a,b,c,na,nb,nc);get(a,c,b,na,nc,nb);
        get(b,a,c,nb,na,nc);get(b,c,a,nb,nc,na);
        get(c,a,b,nc,na,nb);get(c,b,a,nc,nb,na);
        cout<<ans<<endl;
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值