1. 程式人生 > >二分套二分

二分套二分

描述

小Hi的學校正面臨著廢校的大危機。面對學校的危機,小Hi同學們決定從ABC三個班中各挑出一名同學成為偶像。  

成為偶像團體的條件之一,就是3名團員之間的身高差越小越好。  

已知ABC三個班同學的身高分別是A1..AN, B1..BM 和 C1..CL。請你從中選出3名同學Ai, Bj, Ck使得D=|Ai-Bj|+|Bj-Ck|+|Ck-Ai|最小。

輸入

第一行包含3個整數,N, M和L。  

第二行包含N個整數,A1, A2, ... AN。(1 <= Ai <= 100000000)

第三行包含M個整數,B1, B2, ... BM。(1 <= Bi <= 100000000)

第四行包含L個整數,C1

, C2, ... CL。(1 <= Ci <= 100000000)

對於30%的資料, 1 <= N, M, L <= 100  

對於60%的資料,1 <= N, M, L <= 1000  

對於100%的資料,1 <= N, M, L <= 100000

輸出

輸出最小的D。

#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> P;
int a[100010],b[100010],c[100010];
vector<P>vec;
vector<int>ans;

int main()
{
    int n,m,l;
    scanf("%d%d%d",&n,&m,&l);
    for(int i=0;i<n;i++)
        scanf("%d",&a[i]);
    sort(a,a+n);
    for(int i=0;i<m;i++)
        scanf("%d",&b[i]);
    sort(b,b+m);
    for(int i=0;i<l;i++)
        scanf("%d",&c[i]);
    sort(c,c+l);
    for(int i=0;i<n;i++)
    {
        int pos=lower_bound(b,b+m,a[i])-b;
        if(pos>=0&&pos<m)
        {
            if(b[pos]>=a[i])
            {
                vec.push_back(P(a[i],b[pos]));
            }
            if(pos-1>=0)
            {
                vec.push_back(P(b[pos-1],a[i]));
            }
        }
//        if(pos<0)
//        {
//            vec.push_back(P(a[i],b[0]));
//        }
//        if(pos>=m)
//        {
//            vec.push_back(P(b[m-1],a[i]));
//        }
    }
    for(int i=0;i<vec.size();i++)
    {
        int L=vec[i].first;
        int R=vec[i].second;
        //printf("%d %d\n",L,R);
        int pos=lower_bound(c,c+l,L)-c;
        if(pos>=0&&pos<l)
        {
            if(c[pos]>=L&&c[pos]<=R)
                ans.push_back((R-L)*2);
            else if(c[pos]>R)
                ans.push_back((c[pos]-L)*2);
        }
//        else if(pos<0)
//        {
//            if(c[0]<=R)
//                ans.push_back((R-L)*2);
//            else
//                ans.push_back((c[0]-L)*2);
//        }
//        else if(pos>=l)
//        {
//            ans.push_back((R-c[l-1])*2);
//        }
        pos=lower_bound(c,c+l,R)-c;
        if(pos>=0&&pos<l)
        {
            ans.push_back((c[pos]-L)*2);
            if(pos-1>=0)
            {
                if(c[pos-1]>=L)
                    ans.push_back((R-L)*2);
                else
                    ans.push_back((R-c[pos-1])*2);
            }
        }
//        else if(pos<0)
//        {
//            if(c[0]>=L)
//                ans.push_back((c[0]-L)*2);
//        }
//        else if(pos>=l)
//        {
//            if(c[l-1]>=L)
//                ans.push_back((R-L)*2);
//            else
//                ans.push_back((R-c[l-1])*2);
//        }
    }
    sort(ans.begin(),ans.end());
    printf("%d\n",ans[0]);
    return 0;
}
精簡版
#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> P;
int a[100010],b[100010],c[100010];
vector<P>vec;vector<int>ans;
int main()
{
    int n,m,l;scanf("%d%d%d",&n,&m,&l);
    for(int i=0;i<n;i++)scanf("%d",&a[i]);sort(a,a+n);
    for(int i=0;i<m;i++)scanf("%d",&b[i]);sort(b,b+m);
    for(int i=0;i<l;i++)scanf("%d",&c[i]);sort(c,c+l);
    for(int i=0;i<n;i++)
    {
        int pos=lower_bound(b,b+m,a[i])-b;
        if(pos>=0&&pos<m)
        {
            if(b[pos]>=a[i])vec.push_back(P(a[i],b[pos]));
            if(pos-1>=0)vec.push_back(P(b[pos-1],a[i]));
        }
    }
    for(int i=0;i<vec.size();i++)
    {
        int L=vec[i].first;int R=vec[i].second;
        int pos=lower_bound(c,c+l,L)-c;
        if(pos>=0&&pos<l)
        {
            if(c[pos]>=L&&c[pos]<=R)ans.push_back((R-L)*2);
            else if(c[pos]>R)ans.push_back((c[pos]-L)*2);
        }
        pos=lower_bound(c,c+l,R)-c;
        if(pos>=0&&pos<l)
        {
            ans.push_back((c[pos]-L)*2);
            if(pos-1>=0)
            {
                if(c[pos-1]>=L)ans.push_back((R-L)*2);
                else ans.push_back((R-c[pos-1])*2);
            }
        }
    }
    sort(ans.begin(),ans.end());printf("%d\n",ans[0]);
    return 0;
}