题目描述
一个快递公司希望在一条街道建立新的服务中心。公司统计了该街道中所有区域在地图上的位置,并希望能够以此为依据为新的服务中心选址:使服务中心到所有区域的距离的总和最小。
给你一个数组positions,其中positions[i] = [left, right] 表示第 i 个区域在街道上的位置,其中left代表区域的左侧的起点,right代表区域的右侧终点,假设服务中心的位置为location:
如果第 i 个区域的右侧终点right满足 right < location,则第 i 个区域到服务中心的距离为 location - right;
如果第 i 个区域的左侧起点left 满足 left > location,则第 i 个区域到服务中心的距离为left - location;
如果第 i 个区域的两侧left,right满足left <= location <= right,则第 i 个区域到服务中心的距离为0
选择最佳的服务中心位置为location,请返回最佳的服务中心位置到所有区域的距离总和的最小值。
输入描述
先输入区域数组positions的长度n(1 ≤ n ≤ 10^5)
接下来 n 行每行输入成对的left和right值,以空格隔开
-10^9 <left ≤ 10^9
-10^9 <right ≤ 10^9
输出描述
输出为location
输入 | 3 1 2 3 4 10 20 |
输出 | 8 |
输入 | 6 1 3 4 9 2 15 6 27 15 17 5 8 |
输出 | 12 |
输入 | 16 41 67 0 34 24 69 58 78 62 64 5 45 27 81 61 91 42 95 27 36 4 91 2 53 82 92 16 21 18 95 26 47 |
输出 | 127 |
题目解析
此题如果用暴力搜索肯定是超时的
从直觉上来说,肯定是中间位置取得最小值
仔细观察可以发现,当服务中心在两个街区外的中间时,两个街区的距离之和是一个定值。如下图蓝色线,当中心x在B点、C点之间时,AB到x的距离与CD到x的距离之和是一个定值,结果等于BC的距离。而越过这个范围,其值就会升高。
当我们再加入两个更靠近的街区EF和GH,这两个街区到x的距离同样出现前述现象(红色线)。而将两者加起来的结果是棕色线。棕色线就是我们求的距离之和。
从这推论可以发现,假设这些街区可以划分为之间没有重叠的两个集合,这些街区的最低距离就在这两个集合之间。
此时就有两个问题
1.怎么保证最小值一定会落在这个区间?而且其实EF可以和CD配对啊?
这其实是一种贪心算法,所以要让BC覆盖FG。
解决方法就是按照街区的右端点和左端点分别排序,
在配对的时候保证选择的街区右端点F一定在B之右,G和C的关系同理
2.其他不成对的街区就不会干扰吗?
仔细考虑,街区如果不成对的原因要么是最后剩下的街区全部在某一部分共同重叠(不然如果一个街区有两个以上分离开的部分被”其他街区“重叠,就意味着这些”其他街区“可以配对,因为重叠区域的边界一定是某个街区的边界),要么是只剩一个街区没法配队。由于题目说x所在的街区与x的距离为0,只要把x移到这个重叠或者单独街区上就可以无视掉这些街区。
import sys
#输入数据
data=[]
n=int(sys.stdin.readline())#用input()也没关系
for i in range(n):
data.append(list(map(int,sys.stdin.readline().split())))
#将左端点和右端点各自排序
left_sorted_index=sorted(list(range(n)),key=lambda x:data[x][0])
right_sorted_index=sorted(list(range(n)),key=lambda x:data[x][1])
ans=0
i=0
j=n-1
#配队
while data[right_sorted_index[i]][1]<data[left_sorted_index[j]][0] and i<j:
ans=ans+data[left_sorted_index[j]][0]-data[right_sorted_index[i]][1]
i+=1
j-=1
print(ans)