hihoCoder - 1514 : 偶像的条件

1514 : 偶像的条件

时间限制:10000ms
单点时限:1000ms
内存限制:256MB

描述

小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。
样例输入

3 3 3  
170 180 190  
195 185 175  
180 160 200

样例输出

10

思路：
先每个班排序，
然后枚举每个班的每个人，
在另外两个班找到接近当前人的身高即可
。

代码:

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<map>
#include<iostream>
#include<cstring>
using namespace std;
const int INF = 100000 + 4;
int a[INF],b[INF],c[INF];

int main()
{
    int n,m,k;
    while(~scanf("%d%d%d",&n,&m,&k)) {
        memset(a, 0, sizeof(a));
        memset(b, 0, sizeof(b));
        memset(c, 0, sizeof(c));
        for(int i=1; i<=n; i++) {
            scanf("%d",&a[i]);
        }
        for(int i=1; i<=m; i++) {
            scanf("%d",&b[i]);
            m2[b[i]]=1;
        }
        for(int i=1; i<=k; i++) {
            scanf("%d",&c[i]);
            m3[c[i]]=1;
        }
        sort(a+1,a+1+n);
        sort(b+1,b+1+m);
        sort(c+1,c+1+k);
        int res = 2147483647;
        for(int i = 1; i <=n; i++) {
            int loc_b_max = lower_bound(b + 1, b + n + 1, a[i]) - b;
            int loc_c_max = lower_bound(c + 1, c + n + 1, a[i]) -c;
            int loc_b_min = 1;
            int loc_c_min = 1;
            if(loc_b_max > 1) {
                  loc_b_min = loc_b_max - 1;
            }
            if(loc_c_max > 1) {
                loc_c_min = loc_c_max - 1;
            }
            res = min(abs(b[loc_b_max] - a[i]) + abs(c[loc_c_max] - a[i]) + abs(b[loc_b_max] - c[loc_c_max]), res);
            res = min(abs(b[loc_b_max] - a[i]) + abs(c[loc_c_min] - a[i]) + abs(b[loc_b_max] - c[loc_c_min]), res);
            res = min(abs(b[loc_b_min] - a[i]) + abs(c[loc_c_max] - a[i]) + abs(b[loc_b_min] - c[loc_c_max]), res);
           // cout<<res<<endl;
        }
        for(int i = 1; i <=m; i++) {
            int loc_a_max = lower_bound(a + 1, a + m + 1, b[i]) - a;
            int loc_c_max = lower_bound(c + 1, c + m + 1, b[i]) -c;
            int loc_a_min = 1;
            int loc_c_min = 1;
            if(loc_a_max > 1) {
                  loc_a_min = loc_a_max - 1;
            }
            if(loc_c_max > 1) {
                loc_c_min = loc_c_max - 1;
            }
            res = min(abs(a[loc_a_max] - b[i]) + abs(c[loc_c_max] - b[i]) + abs(a[loc_a_max] - c[loc_c_max]), res);
            res = min(abs(a[loc_a_max] - b[i]) + abs(c[loc_c_min] - b[i]) + abs(a[loc_a_max] - c[loc_c_min]), res);
            res = min(abs(a[loc_a_min] - b[i]) + abs(c[loc_c_max] - b[i]) + abs(a[loc_a_min] - c[loc_c_max]), res);
           // cout<<res<<endl;
        }
        for(int i = 1; i <=k; i++) {
            int loc_b_max = lower_bound(b + 1, b + k + 1, c[i]) - b;
            int loc_a_max = lower_bound(a + 1, a + k + 1, c[i]) -a;
            int loc_b_min = 1;
            int loc_a_min = 1;
            if(loc_b_max > 1) {
                  loc_b_min = loc_b_max - 1;
            }
            if(loc_a_max > 1) {
                loc_a_min = loc_a_max - 1;
            }
            res = min(abs(b[loc_b_max] - c[i]) + abs(a[loc_a_max] - c[i]) + abs(b[loc_b_max] - a[loc_a_max]), res);
            res = min(abs(b[loc_b_max] - c[i]) + abs(a[loc_a_min] - c[i]) + abs(b[loc_b_max] - a[loc_a_min]), res);
            res = min(abs(b[loc_b_min] - c[i]) + abs(a[loc_a_max] - c[i]) + abs(b[loc_b_min] - a[loc_a_max]), res);
           // cout<<res<<endl;
        }
        printf("%d\n", res);
    }
    return 0;
}
/*
3 3 3
1705 180 190
175 185 195
160 180 200
1 1 1
100000000
1
100000000
2 2 2
1 2
2 34
5 2
3 3 3
4 4 4
1 3 7
4 4 4
*/