我的程序应该计算最近对的距离。它接受两个排序数组: xpts 是按 x 坐标排序的对/坐标数组。ypts 是一个按 y 坐标排序的数组。我正在尝试使用分而治之的技术,所以我递归地传入一半的数组。但是,我得到一个错误。
TypeError: 'float' object is not iterable: a1, pair1= closest_pair(xlft,ylft)
我的代码是:
def closest_pair(xpts,ypts):
if xpts size < = 3:
if xsize==1:
return xpts[0][0]
elif xsize==2:
return dist(xpts[0],xpts[1])
else:
one= xpts[0]
two= xpts[1]
three= xpts[2]
s1= dist(one,two)
s2= dist(two,three)
s3= dist(one,three)
s= (min(s1,s2,s3),min(xpts[0],xpts[1],xpts[2]))
return s
else:
...
xlft= xpts[:xsize/2]
xrht= xpts[(xsize/2)+1:]
ylft= []
yrht= []
median= xpts[(xsize/2)-1][0]
for p in ypts:
if p[0] <= median:
ylft.append(p)
else:
yrht.append(p)
a1, pair1= closest_pair(xlft,ylft)
a2, pair2= closest_pair(xrht,yrht)
st= []
if a1 < a2:
a3, pair3= (a1,pair1)
else:
a3, pair3= (a2,pair2)
for p in ypts:
if abs(p[0]-median) < a3:
st.append(p)
n_st= len(st)
closest= (a3,pair3)
if n_st>1:
for i in range(n_st-1):
for j in range(i+1,min(i+8,n_st)):
if dist(st[i],st[j]) < closest[0]:
closest= (dist(st[i],st[j]),(st[i],st[j]))
d= closest
return d
d1 = closest_pair(xpts, ypts)[0]
print d1