校招真题练习008 浇花(百度)

浇花

题目描述
一个花坛中有很多花和两个喷泉。
喷泉可以浇到以自己为中心，半径为r的圆内的所有范围的花。
现在给出这些花的坐标和两个喷泉的坐标，要求你安排两个喷泉浇花的半径r1和r2，
使得所有的花都能被浇到的同时, r1^2 + r2^2 的值最小。

输入描述:
第一行5个整数n，x1，y1，x2，y2表示花的数量和两个喷泉的坐标。
接下来n行，每行两个整数xi, yi表示第i朵花的坐标。
满足1 <= n <= 2000，花和喷泉的坐标满足-107<= x, y <= 107。
输出描述:
一个整数，r1^2 + r2^2 的最小值。

import sys
lines = sys.stdin.readlines()
#print(lines)
line0 = list(map(int,lines[0].strip().split()))
n = line0[0]
x1,y1,x2,y2 = line0[1],line0[2],line0[3],line0[4]
#print(x1,y1,x2,y2)
flowers = []
line_next = lines[1:]
for i in range(len(line_next)):
    line = list(map(int,line_next[i].strip().split()))
    flowers.append([line[0],line[1]])
#print(flowers)
dic1 = {-1:0}
dic2 = {-1:0}
minR1 = 0
minR2 = 0

for i in range(len(flowers)):
    distance1 = (flowers[i][0]-x1)**2 + (flowers[i][1]-y1)**2
    distance2 = (flowers[i][0]-x2)**2 + (flowers[i][1]-y2)**2
    dic1[i] = distance1
    minR1 = max(minR1,distance1)
    dic2[i] = distance2
    minR2 = max(minR2,distance2)
minR = min(minR1,minR2)
#print(dic1,dic2)
lst1 = sorted(dic1.items(),key=lambda x:x[1])
t1 = minR1
t2 = minR2
for idx in range(1,len(lst1)):
    flowerid = lst1[idx][0]
    t1 = lst1[idx][1]
    dic2.pop(flowerid)
    t2 = min(t2,max(dic2.values()))
    minR = min(minR,t1+t2)
print(minR)

类型：二维数组，排序

思路：贪心算法，先将所有的花与某一个喷头的距离排序，按距离从近到远对此花进行分配。

如果分配到喷头1的范围，则喷头一的r会逐渐增加，同时将此花从喷头二的列表中删除，重新计算喷头二的最大值。