20年408真题 三元组的最小距离

20年408真题 三元组的最小距离

定义三元组（a,b,c）(a,b,c均为整数)的距离D = ∣ a − b ∣ + ∣ b − c ∣ + ∣ c − a ∣ 。给定3个非空整数集合S1，S2和S3，按升序分别存储在3个数组中。请设计一种尽可能高效的算法，计算并输出所有可能的三元组(a,b,c)(a ∈ S1 , b ∈ S2 , c ∈ S3)中的最小距离。

阅读完题目，很容易就能想到三重循环O（n^3) 暴力计算的方法。但是这样时间复杂度过高了，需要进行优化 。

算法思路：
D = ∣ a − b ∣ + ∣ b − c ∣ + ∣ c − a ∣ = 2 * ∣ c − a ∣
D的距离是最大值和最小值的距离（也就是最大值和最小值之差乘2）
所以只要求出当前三元组的最小值和最大值就可以算出当前距离
首先定义一个极大值变量，三个数组的长度以此为lena,lenb,lenc,可以通过三指针方法，求出当前三元组的最小值，在更新变量，谁是最小值就移动下标指针，直至遍历完成 ，最后得出的变量就是我们所求的最小的∣ c − a ∣，乘2就是所求的三元组的最小距离。
时间复杂度O(lena+lenb+lenc)

代码示例：

int min(int a,int b){
    return a > b ? b : a;
}

int abs(int a){
    return a < 0 ? -a : a;
}

int MIN(int a,int b,int c){
    if(a <= b && a <= c) return a;
    else if(b <= a && b <= c) return b;
    else return c;
}

int MAX(int a,int b,int c){
    if(a >= b && a >= c) return a;
    else if(b >= a && b >= c) return b;
    else return c;
}


int solve(int *a,int lena,int *b,int lenb,int *c,int lenc){
    int ans = 0x7fffffff;
    int i = 0, j = 0,k = 0;
    while(i < lena && j < lenb && k < lenc){
        ans = min(ans,MAX(a[i],b[j],c[k])-MIN(a[i],b[j],c[k]));
        if(a[i] <= b[j] && a[i] <= c[k]) i++;
        else if(b[j] <= a[i] && b[j] <= c[k]) j++;
        else k++;
    }
    //printf("%d\n",ans);
    return ans * 2;
}