蓝哥算法工程第二期_kk

 

算法思路：

从小到大枚举S2数组中的每个元素作为中间值b。对于每个中间值b，在S1数组中二分查找到第一个大于等于b的元素a，在S3数组中二分查找到第一个大于等于b的元素c，此时，要么a、b、c都在各自的数组中取到了最大或最小值，要么至少有一个元素可以向前或向后移动一位，得更小的距离，比较以下四个值：D1=|a-b|+|b-c|+|c-a|，D2=|a-b|+|b-(c-1)|+|(c-1)-a|，D3=|a-b|+|b-(c+1)|+|(c+1)-a|，D4=|a-(b-1)|+|(b-1)-c|+|c-a|，D5=|a-(b+1)|+|(b+1)-c|+|c-a|，并更新最小距离。

算法实现：
 

def min_distance(s1, s2, s3):
    min_d = float("inf")
    for b in s2:
        i = bisect.bisect_left(s1, b)
        j = bisect.bisect_left(s3, b)
        for a in (s1[i-1:i] if i>0 else [s1[0]]):
            for c in (s3[j:j+1] if j<len(s3) else [s3[-1]]):
                d1 = abs(a-b) + abs(b-c) + abs(c-a)
                d2 = abs(a-b) + abs(b-(c-1)) + abs((c-1)-a)
                d3 = abs(a-b) + abs(b-(c+1)) + abs((c+1)-a)
                d4 = abs((a-1)-b) + abs(b-c) + abs(c-(a-1))
                d5 = abs((a+1)-b) + abs(b-c) + abs(c-(a+1))
                d = min(d1, d2, d3, d4, d5)
                min_d = min(min_d, d)

算法分析：

时间复杂度：假设三个数组的长度分别为n1、n2、n3，枚举S2数组的元素个数为n2，对于每个元素，需要进行两次二分查找和常数次操作，因此总时间复杂度为O(logn2)。

空间复杂度：该算法只需要常数个辅助变量和调用bisect库所需的空间，因此空间复杂度为O(1)。